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-20 01:16:38 +09:00
parent 7c12e655b4
commit 31394e7694
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Simulation.py
View File

@@ -179,11 +179,11 @@ if __name__ == "__main__":
# to check bying # to check bying
stock_codes = { stock_codes = {
#"252670": ['20220602'], "252670": ['20220811'],
"122630": ['20220611', '20220612', '20220613', '20220614'], #"122630": ['20220611', '20220612', '20220613', '20220614'],
} }
method = "rule" # "rule", "ml", "answer" method = "ml" # "rule", "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

@@ -236,5 +236,5 @@ if __name__ == "__main__":
stock_code = "252670" stock_code = "252670"
vitTrainer = VitTrainer(RESOURCE_PATH) vitTrainer = VitTrainer(RESOURCE_PATH)
train_ds, val_ds = vitTrainer.getData(stock_code, sDate="20220809", eDate="20220812") train_ds, val_ds = vitTrainer.getData(stock_code, sDate="20220701", eDate="20220731")
vitTrainer.train(train_ds, val_ds, model_path) vitTrainer.train(train_ds, val_ds, model_path)