etna icon indicating copy to clipboard operation
etna copied to clipboard

MLP model

Open martins0n opened this issue 2 years ago • 0 comments

🚀 Feature Request

Add MLPModel model for etna deep model family

Proposal

class MLPNet(DeepBaseNet):
    def __init__(
        self,
        input_size: int,
        hidden_size: List[int],
        lr: float,
        loss: "torch.nn.Module",
        optimizer_params: Optional[dict],
    ) -> None:
   pass
  • For MLP we dont need encoder part: so encoder_lenght=0
  • Typical batch sample would have shape [encoder_lenght, input_size]

Test cases

  • add unit tests for make_sample
  • add seasonal overfit test
  • add test for step and retrun values
  • add test for forward and return values

Additional context

No response

martins0n avatar Aug 01 '22 10:08 martins0n