.

.gitignore

@@ -1,2 +1,3 @@
+
+artifacts/
 __pycache__/
-artifacts/

train.py

@@ -76,7 +76,7 @@ def train(symbol: str, seq_len: int = 60, epochs: int = 5, batch_size: int = 32,
     model = StockLSTM(input_dim=1, hidden_dim=64, num_layers=2, dropout=0.2).to(device)
     criterion = nn.MSELoss()
     # optimizer = optim.Adam(model.parameters(), lr=lr)
-    optimizer = optim.SGD(model.parameters(), lr=lr)
+    optimizer = optim.Adam(model.parameters(), lr=lr)
 
     # --- training ---
     model.train()