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

init

Browse Source
This commit is contained in:
dsyoon
2022-08-07 13:38:07 +09:00
parent c173a6d7dc
commit 655eedc421
3 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
Simulation.py
View File

@@ -28,6 +28,9 @@ class Simulation (HTS):
return return
def draw(self, stock_code, given_day, data, bsLine): def draw(self, stock_code, given_day, data, bsLine):
if bsLine is None:
return
# 어제 데이터는 지운다. # 어제 데이터는 지운다.
data = data.loc[pd.DatetimeIndex(data.index).day == int(given_day[6:])] data = data.loc[pd.DatetimeIndex(data.index).day == int(given_day[6:])]
buy_line = bsLine['buy'][381:] buy_line = bsLine['buy'][381:]
@@ -144,7 +147,7 @@ class Simulation (HTS):
X, Y = self.stock2Vector.getDataset2D(data) X, Y = self.stock2Vector.getDataset2D(data)
predY = self.stockPredictor.predict(X, Y) predY = self.stockPredictor.predict(X, Y)
print (predY) bsLine = None
else: else:
LAST_DATA = self.stock2Vector.getLastData(stock_code, today) LAST_DATA = self.stock2Vector.getLastData(stock_code, today)
result = self.stock2Vector.getRealTime(stock_code, today, LAST_DATA) result = self.stock2Vector.getRealTime(stock_code, today, LAST_DATA)
@@ -169,10 +172,10 @@ if __name__ == "__main__":
stock_codes = { stock_codes = {
# 252670 # 252670
# 122630 # 122630
"252670": ['20200731'], "252670": ['20220801', '20220802', '20220803', '20220804', '20220805'],
} }
method = "ml" # "ml", "answer" method = "rul" # "rul", "ml", "answer"
for stock_code in stock_codes: for stock_code in stock_codes:
simulation = Simulation(RESOURCE_PATH) simulation = Simulation(RESOURCE_PATH)

2
VitTrainer.py
View File

@@ -52,7 +52,7 @@ class VitTrainer:
metric_for_best_model="accuracy", metric_for_best_model="accuracy",
logging_dir='logs', logging_dir='logs',
remove_unused_columns=False, remove_unused_columns=False,
num_train_epochs=20, num_train_epochs=14,
) )
return return

2
stock/util/StockPredictor.py
View File

@@ -24,7 +24,7 @@ class StockPredictor:
def __init__(self, RESOURCE_PATH): def __init__(self, RESOURCE_PATH):
self.RESOURCE_PATH = RESOURCE_PATH self.RESOURCE_PATH = RESOURCE_PATH
self.model_dir = os.path.join(RESOURCE_PATH, "tmp") self.model_dir = os.path.join(RESOURCE_PATH, "model")
self.stock2Vector = Stock2Vector(RESOURCE_PATH) self.stock2Vector = Stock2Vector(RESOURCE_PATH)
self.set_seed(42) self.set_seed(42)