It is unclear how to parameterize MLPModel parameter hidden_size because it accepts list of sizes for the layers. Even if we come up with parametrization it is unclear how to implement it because it is expected that we have a complete set of parameter values. We can use CategoricalDistribution but it seems not very effective.
🚀 Feature Request
Add
params_to_tune
for new NNs (rnn, mlp).Proposal
Sources:
Suggested grid for
RNNModel
:Suggested grid for
MLPModel
:To discuss:
lr
necessary?MLPModel
parameterhidden_size
because it accepts list of sizes for the layers. Even if we come up with parametrization it is unclear how to implement it because it is expected that we have a complete set of parameter values. We can useCategoricalDistribution
but it seems not very effective.Test cases
Look at https://github.com/tinkoff-ai/etna/issues/1184.
Additional context
No response