tinkoff-ai / etna

ETNA – Time-Series Library
https://etna.tinkoff.ru
Apache License 2.0
855 stars 81 forks source link

Add `params_to_tune` for new NNs (`RNNModel`, `MLPModel`) #1196

Closed Mr-Geekman closed 1 year ago

Mr-Geekman commented 1 year ago

🚀 Feature Request

Add params_to_tune for new NNs (rnn, mlp).

Proposal

Sources:

Suggested grid for RNNModel:

{
    "num_layers": IntUniformDistribution(1, 3, step=1),
    "hidden_size": IntUniformDistribution(4, 64, step=4),
    "lr": LogUniformDistribution(1e-5, 1e-2),
}

Suggested grid for MLPModel:

{
    "num_layers": IntUniformDistribution(1, 3, step=1),
    "hidden_size": IntUniformDistribution(4, 64, step=4),
    "lr": LogUniformDistribution(1e-5, 1e-2),
}

To discuss:

Test cases

Look at https://github.com/tinkoff-ai/etna/issues/1184.

Additional context

No response

martins0n commented 1 year ago
Mr-Geekman commented 1 year ago

Closed by #1218.