search

tidymodels / brulee

High-Level Modeling Functions with 'torch'
https://brulee.tidymodels.org/
Other
67 stars 7 forks source link

Two layer neural networks #82

Closed topepo closed 1 week ago

topepo commented 1 week ago

Added a convenience function, brulee_mlp_two_layer(), to more easily fit two-layer networks with parsnip.

Relies on two new dials parameters. Also, tunable information is moved to parsnip (in a PR after this is finalized).