Machine Learning with R Cookbook - Second Edition

How to do it...

Perform the following steps to train a neural network with neuralnet:

1. First, load the iris dataset and split the data into training and testing datasets:

    > data(iris)
    > ind = sample(2, nrow(iris), replace = TRUE, prob=c(0.7, 0.3))
    > trainset = iris[ind == 1,]
    > testset = iris[ind == 2,]

2. Then, install and load the neuralnet package:

    > install.packages("neuralnet")
    > library(neuralnet)

3. Add the columns versicolor, setosa, and virginica based on the name matched value in the Species column:

    > trainset$setosa = trainset$Species == "setosa"
    > trainset$virginica = trainset$Species == "virginica"
    > trainset$versicolor = trainset$Species == "versicolor"

4. Next, train the neural network with the neuralnet function with three ...