init

VitTrainer.py

@@ -1,6 +1,4 @@
 # tensor - numpy - PILImage 변환 (https://qlsenddl-lab.tistory.com/37)
-from PIL import Image
-
 import os
 os.environ['KMP_DUPLICATE_LIB_OK']='True'
 import random
@@ -192,7 +190,8 @@ class VitTrainer:
 
     def getData(self, stock_code, sDate, eDate):
         data = self.stock2Vector.getTrainData(stock_code, sDate, eDate)
-        X, Y = self.stock2Vector.getDataset2D(data)
+        #X, Y = self.stock2Vector.getDataset2D(data)
+        X, Y = self.stock2Vector.getVectorData(data)
         print("Data count: ", len(X))
 
         trans = transforms.ToPILImage()
@@ -215,8 +214,6 @@ class VitTrainer:
 
 if __name__ == "__main__":
 
-    image = Image.open("img.png")
-
     PROJECT_HOME = os.getcwd()
     RESOURCE_PATH = os.path.join(PROJECT_HOME, "resources")
     model_path = os.path.join(RESOURCE_PATH, "model")