dsyoon/DeepStock
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

init

Browse Source
This commit is contained in:
dosangyoon
2022-08-07 17:53:10 +09:00
parent 69c52b5e1b
commit 0796547466
2 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
VitTrainer.py
View File

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

2
stock/util/Stock2Vector.py
View File

@@ -285,7 +285,7 @@ class Stock2Vector(HTS):
return X, Y return X, Y
def getVectorData(self, data, type="avg10", VECTOR_SIZE = 32): 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) df, minmax_df = self.preprocessData(data)
X = np.asarray(X) X = np.asarray(X)