traineR (version 2.2.11)

train.nnet: train.nnet

Description

Provides a wrapping function for the nnet.

Usage

train.nnet(formula, data, weights, ..., subset, na.action, contrasts = NULL)

Value

A object nnet.prmdt with additional information to the model that allows to homogenize the results.

Arguments

formula

A formula of the form class ~ x1 + x2 + ...

data

Data frame from which variables specified in formula are preferentially to be taken.

weights

(case) weights for each example – if missing defaults to 1.

...

arguments passed to or from other methods.

subset

An index vector specifying the cases to be used in the training sample. (NOTE: If given, this argument must be named.)

na.action

A function to specify the action to be taken if NAs are found. The default action is for the procedure to fail. An alternative is na.omit, which leads to rejection of cases with missing values on any required variable. (NOTE: If given, this argument must be named.)

contrasts

a list of contrasts to be used for some or all of the factors appearing as variables in the model formula.

See Also

The internal function is from package nnet.

Examples

# Classification
data("iris")

n <- seq_len(nrow(iris))
.sample <- sample(n, length(n) * 0.75)
data.train <- iris[.sample,]
data.test <- iris[-.sample,]

modelo.nn <- train.nnet(Species~., data.train, size = 20)
modelo.nn
prob <- predict(modelo.nn, data.test, type = "prob")
prob
prediccion <- predict(modelo.nn, data.test, type = "class")
prediccion

# Regression
len <- nrow(swiss)
sampl <- sample(x = 1:len,size = len*0.20,replace = FALSE)
ttesting <- swiss[sampl,]
ttraining <- swiss[-sampl,]
model.knn <- train.nnet(Infant.Mortality~.,ttraining, size = 20)
prediction <- predict(model.knn, ttesting)
prediction