Poetry dependency & build manager

.pre-commit-config.yaml

@@ -17,3 +17,9 @@ repos:
       - id: end-of-file-fixer
       - id: mixed-line-ending
       - id: trailing-whitespace
+
+  - repo: https://github.com/psf/black
+    rev: 24.4.0
+    hooks:
+      - id: black
+        args: ["--line-length=120"]

README.md

@@ -253,4 +253,4 @@ study.best_params  # E.g. {'x': 2.002108042}
 ---
 ## Sources:
 
-- [github.com/leggedrobotics/plr-exercise](https://github.com/leggedrobotics/plr-exercise) by @JonasFrey96
+- [github.com/leggedrobotics/plr-exercise](https://github.com/leggedrobotics/plr-exercise) by @JonasFrey96

model.py

@@ -0,0 +1,27 @@
+from torch import nn
+
+
+class Model_Linear(nn.Module):
+
+    def __init__(self) -> None:
+
+        super(Model_Linear, self).__init__()
+
+        self.encoder = nn.Sequential(
+            nn.Flatten(),
+            nn.Linear(28 * 28, 128),
+            nn.BatchNorm1d(128),
+            nn.ReLU(),
+            nn.Linear(128, 60),
+        )
+
+        self.decoder = nn.Sequential(
+            nn.Linear(60, 128), nn.BatchNorm1d(128), nn.ReLU(), nn.Linear(128, 28 * 28), nn.Sigmoid()
+        )
+
+    def forward(self, x):
+
+        encoded = self.encoder(x)
+        decoded = self.decoder(encoded)
+
+        return decoded