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 04:25:58 +09:00
parent 19644cd7bc
commit df52c7bd8b
4 changed files with 54 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
stock/util/LabelMaker.py
View File

@@ -289,8 +289,9 @@ class LabelMaker (HTS):
bsLine['buy'] = [-1 for i in range(size)]
bsLine['sell'] = [-1 for i in range(size)]
for i in range(size):
self.buySellChecker.write(outFp, data, i)
outFp.flush()
if outFp is not None:
self.buySellChecker.write(outFp, data, i)
outFp.flush()
if labels[i] == 1:
bsLine['sell'][i] = closes[i]
if labels[i] == 2:
@@ -313,23 +314,40 @@ if __name__ == "__main__":
UPDATE_MODE = False
if UPDATE_MODE:
stock_code = "122630"
ymd = '20220812'
"""
# 매일 입력하면서 정답 셋 만들기
stock_code = "252670"
ymd = '20220730'
labelMaker.clearLabel(db_filename, stock_code, ymd)
labelMaker.makeLabel(db_filename, stock_code, ymd, "0910", 2)
labelMaker.makeLabel(db_filename, stock_code, ymd, "0911", 2)
labelMaker.makeLabel(db_filename, stock_code, ymd, "0912", 2)
labelMaker.makeLabel(db_filename, stock_code, ymd, "0915", 1)
labelMaker.makeLabel(db_filename, stock_code, ymd, "0916", 1)
labelMaker.makeLabel(db_filename, stock_code, ymd, "0948", 2)
"""
labelMaker.makeLabel(db_filename, stock_code, ymd, "0933", 2)
labelMaker.makeLabel(db_filename, stock_code, ymd, "0934", 2)
labelMaker.makeLabel(db_filename, stock_code, ymd, "0935", 2)
labelMaker.makeLabel(db_filename, stock_code, ymd, "0936", 2)
labelMaker.makeLabel(db_filename, stock_code, ymd, "1004", 1)
labelMaker.makeLabel(db_filename, stock_code, ymd, "1005", 1)
labelMaker.makeLabel(db_filename, stock_code, ymd, "1006", 1)
labelMaker.makeLabel(db_filename, stock_code, ymd, "1007", 1)
labelMaker.makeLabel(db_filename, stock_code, ymd, "1412", 1)
labelMaker.makeLabel(db_filename, stock_code, ymd, "1413", 1)
labelMaker.makeLabel(db_filename, stock_code, ymd, "1414", 1)
labelMaker.makeLabel(db_filename, stock_code, ymd, "1415", 1)
labelMaker.makeLabel(db_filename, stock_code, ymd, "1416", 1)
labelMaker.makeLabel(db_filename, stock_code, ymd, "1417", 1)
labelMaker.showLabels(stock_code, ymd)
else:
stock_codes = {
"252670": ['20220801', '20220802', '20220803', '20220804', '20220805', '20220808', '20220809', '20220810', '20220811', '20220812'],
"122630": ['20220801', '20220802', '20220803', '20220804', '20220805', '20220808', '20220809', '20220810', '20220811', '20220812'],
"252670": ['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'],
#"122630": ['20220801', '20220802', '20220803', '20220804', '20220805', '20220808', '20220809', '20220810', '20220811', '20220812'],
}
view = True
@@ -342,7 +360,7 @@ if __name__ == "__main__":
if view:
labelMaker.showLabels(stock_code, ymd, outFp)
else:
stock_codes = ["252670", "122630"]
# 종목에 대해서 주어진 일자에 대해서 통계치 추출하기
for stock_code in stock_codes:
ymds = labelMaker.getDate(stock_code)
for ymd in ymds: