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-17 16:14:45 +09:00
parent c0111f9d3b
commit 517196141e
2 changed files with 16 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
LabelMaker.py
View File

@@ -1,5 +1,6 @@
import os import os
import logging import logging
logging.basicConfig(level=logging.INFO)
from stock.util.LabelChecker import LabelChecker from stock.util.LabelChecker import LabelChecker
class LabelMaker: class LabelMaker:
@@ -24,6 +25,7 @@ class LabelMaker:
return return
def write(self, stock_code, ymd, outFp): def write(self, stock_code, ymd, outFp):
logging.info(stock_code + " / " + ymd)
self.labelChecker.showLabels(stock_code, ymd, outFp) self.labelChecker.showLabels(stock_code, ymd, outFp)
return return
@@ -32,7 +34,7 @@ class LabelMaker:
for stock_code in stock_codes: for stock_code in stock_codes:
ymds = self.labelChecker.getDate(stock_code) ymds = self.labelChecker.getDate(stock_code)
for ymd in ymds: for ymd in ymds:
logging.info(stock_code, ymd) logging.info(stock_code + " / " + ymd)
bsLine, data = self.labelChecker.makeCandidate(stock_code, ymd) bsLine, data = self.labelChecker.makeCandidate(stock_code, ymd)
self.labelChecker.updateLabel(self.db_filename, stock_code, bsLine, data, ymd) self.labelChecker.updateLabel(self.db_filename, stock_code, bsLine, data, ymd)
return return
@@ -51,11 +53,17 @@ if __name__ == "__main__":
'20220729', '20220801', '20220802', '20220803', '20220804', '20220729', '20220801', '20220802', '20220803', '20220804',
'20220805', '20220808', '20220809', '20220810', '20220811', '20220805', '20220808', '20220809', '20220810', '20220811',
'20220812', '20220816', '20220817'], '20220812', '20220816', '20220817'],
# "122630": ['20220801', '20220802', '20220803', '20220804', '20220805', '20220808', '20220809', '20220810', '20220811', '20220812'], "122630": ['20220701', '20220704', '20220705', '20220706', '20220707',
'20220708', '20220711', '20220712', '20220713', '20220714',
'20220715', '20220718', '20220719', '20220720', '20220721',
'20220722', '20220725', '20220726', '20220727', '20220728',
'20220729', '20220801', '20220802', '20220803', '20220804',
'20220805', '20220808', '20220809', '20220810', '20220811',
'20220812', '20220816', '20220817'],
} }
MODE = "UPDATE" MODE = "WRITE"
if MODE == "UPDATE": if MODE == "UPDATE":
# 매일 입력하면서 정답 셋 만들기 # 매일 입력하면서 정답 셋 만들기
stock_code = "252670" stock_code = "252670"
@@ -82,7 +90,8 @@ if __name__ == "__main__":
elif MODE == "WRITE": elif MODE == "WRITE":
for stock_code in stock_codes: for stock_code in stock_codes:
outFp = open("check_"+stock_code+".txt", "w") fileName = os.path.join(RESOURCE_PATH, "tmp", "check_"+stock_code+".txt")
outFp = open(fileName, "w")
for ymd in stock_codes[stock_code]: for ymd in stock_codes[stock_code]:
labelMaker.write(stock_code, ymd, outFp) labelMaker.write(stock_code, ymd, outFp)
outFp.close() outFp.close()

1
stock/util/LabelChecker.py
View File

@@ -296,6 +296,7 @@ class LabelChecker (HTS):
if labels[i] == 2: if labels[i] == 2:
bsLine['buy'][i] = lows[i] bsLine['buy'][i] = lows[i]
if outFp is None:
#self.draw_simple(stock_code, ymd, data, bsLine) #self.draw_simple(stock_code, ymd, data, bsLine)
self.draw_detail(stock_code, ymd, data, bsLine) self.draw_detail(stock_code, ymd, data, bsLine)
return return