使用带有神经网络的 Caret 包作为方法

Question

I have created a prediction model using caret neuralnet method. Now I'm wondering which is the defualt activation function that caret is using for this method.

Answer 1

The caret package uses theneuralnet package when caret::train() is used with method="neuralnet" , as documented in train models by tag on the caret documentation website .

The specific function called is neuralnet::neuralnet() .

Answer 2

As mentioned by Len Greski method neuralnet calls neuralnet::neuralnet() . The activation function used by neuralnet::neuralnet() is act.fct = "logistic" .

Caret does not change the default which can be observed in the source . Relevant part of code:

neuralnet::neuralnet(form, data = dat, hidden = nodes, ...)

One can also supply a user defined function as part of the ellipsis in caret::train . Relevant answer on SO explaining how to create such a function.