Variable number of layers in the MLP

tidymodels / brulee

High-Level Modeling Functions with 'torch'

https://brulee.tidymodels.org/

Other

67 stars 7 forks source link

Variable number of layers in the MLP #9

Closed dfalbel closed 3 years ago

dfalbel commented 4 years ago

Add an argument called hidden, a vector with the number of hidden units for each hidden layer. The number of hidden layers would be equal to the length(hidden).

topepo commented 3 years ago

We could make both hidden_units and activation vectors.

github-actions[bot] commented 3 years ago

This issue has been automatically locked. If you believe you have found a related problem, please file a new issue (with a reprex: https://reprex.tidyverse.org) and link to this issue.