init

2022-08-13 17:16:47 +09:00
parent 8ec05b8447
commit f9ffa363fa
5 changed files with 100 additions and 126 deletions
--- a/stock/util/StockPredictor.py
+++ b/stock/util/StockPredictor.py
@@ -102,7 +102,7 @@ class StockPredictor:
    def predict(self, X, Y=None):
        print("Data count: ", len(X))

-        X = [self.trans(torch.tensor([x])) for x in X]
+        X = [self.trans(torch.tensor(x)) for x in X]

        test_X = X
        test_Y = Y