From 3258da30151a812f8d8e9dba3c26e36c9e63108d Mon Sep 17 00:00:00 2001 From: Robin Meersman Date: Tue, 9 Dec 2025 21:07:43 +0100 Subject: [PATCH] feat: encoder skeleton code --- models/autoencoder.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 models/autoencoder.py diff --git a/models/autoencoder.py b/models/autoencoder.py new file mode 100644 index 0000000..b66fb73 --- /dev/null +++ b/models/autoencoder.py @@ -0,0 +1,18 @@ +import torch +import torch.nn as nn + + +class Encoder(nn.Module): + def __init__(self, input_size, hidden_size, output_size): + super(Encoder, self).__init__() + + def forward(self, x: torch.Tensor) -> torch.Tensor: + pass + + +class Decoder(nn.Module): + def __init__(self): + super(Decoder, self).__init__() + + def forward(self, x: torch.Tensor) -> torch.Tensor: + pass