shuim/auto-stock
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

백테스트 수정중

Browse Source
This commit is contained in:
shuim
2026-03-07 13:16:01 +09:00
parent ea355eb60c
commit 7c9b596098
13 changed files with 908 additions and 303 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/app/services/ml_ensemble.py
View File

@@ -111,8 +111,8 @@ class StockLSTM:
self.seq_std = X_seq.std(axis=(0, 1)) + 1e-8
X_norm = (X_seq - self.seq_mean) / self.seq_std
X_tensor = torch.FloatTensor(X_norm).to(self.device)
y_tensor = torch.LongTensor(y).to(self.device)
X_tensor = torch.FloatTensor(X_norm.copy()).to(self.device)
y_tensor = torch.LongTensor(y.copy()).to(self.device)
dataset = TensorDataset(X_tensor, y_tensor)
loader = DataLoader(dataset, batch_size=batch_size, shuffle=True)
@@ -157,7 +157,7 @@ class StockLSTM:
self.model.eval()
X_norm = (X_seq - self.seq_mean) / self.seq_std
X_tensor = torch.FloatTensor(X_norm).to(self.device)
X_tensor = torch.FloatTensor(X_norm.copy()).to(self.device)
with torch.no_grad():
outputs = self.model(X_tensor)