search

topepo / caret

caret (Classification And Regression Training) R package that contains misc functions for training and plotting classification and regression models
http://topepo.github.io/caret/index.html
1.6k stars 635 forks source link

Changing activation function in dnn #531

Closed sparcycram closed 7 years ago

sparcycram commented 7 years ago

Is there anyway to change the activationfunction on dnn in caret?

I have tried using activationfun = "linear" from package deepnet in both tuneGrid = expand.grid(...., activationfun = "linear") and in trControl.

Neither worked. Am I doing something wrong or is this functionality not available?

Thanks

topepo commented 7 years ago

The method isn't parameterized for changing that right now. You can use the existing code to create a custom method. Pull requests are always welcome too =]