init

2022-08-07 17:46:06 +09:00
parent 655eedc421
commit 57a906bc08
5 changed files with 53 additions and 26 deletions
--- a/stock/util/Stock2Vector.py
+++ b/stock/util/Stock2Vector.py
@@ -284,6 +284,15 @@ class Stock2Vector(HTS):
        Y = np.asarray(Y, dtype='int64')
        return X, Y

+    def getVectorData(self, data, type="avg10", VECTOR_SIZE = 32):
+        X, Y = [], []
+        df, minmax_df = self.preprocessData(data)
+
+        X = np.asarray(X)
+        Y = np.asarray(Y, dtype='int64')
+        return X, Y
+
+
    def getDataset3D(self, data, VECTOR_SIZE = 299):
        df, minmax_df = self.preprocessData(data)