init

2022-08-07 17:53:10 +09:00
parent 69c52b5e1b
commit 0796547466
2 changed files with 3 additions and 6 deletions
--- a/stock/util/Stock2Vector.py
+++ b/stock/util/Stock2Vector.py
@@ -285,7 +285,7 @@ class Stock2Vector(HTS):
        return X, Y

    def getVectorData(self, data, type="avg10", VECTOR_SIZE = 32):
-        X, Y = [], []
+        X, Y = np.zeros((VECTOR_SIZE, VECTOR_SIZE, 4)), np.zeros((32,1))
        df, minmax_df = self.preprocessData(data)

        X = np.asarray(X)