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

init

Browse Source
This commit is contained in:
dosang.yoon
2022-07-10 16:08:24 +09:00
parent 211474bd1b
commit 5ce3be27ba
4 changed files with 469 additions and 150 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
hts/Simulation.py
View File

@@ -1,5 +1,5 @@
from math import nan
from datetime import datetime, timedelta
from datetime import datetime
import csv
import pandas as pd
import plotly.graph_objects as go
@@ -93,11 +93,11 @@ class Simulation:
sell_check = go.Scatter(x=data['date'], y=sell_line, mode='markers', name="sell", marker=dict(size=14, color=sell_colors, line_width=0))
upper = go.Scatter(x=data['date'], y=data["upper"], name="upper", line_color='#000000')
lower = go.Scatter(x=data['date'], y=data["lower"], name="lower", line_color='#000000')
avg3 = go.Scatter(x=data['date'], y=data["avg3"], name="avg3", line_color='#000000')
avg5 = go.Scatter(x=data['date'], y=data["avg5"], name="avg5", line_color='#085F1B')
avg3 = go.Scatter(x=data['date'], y=data["avg3"], name="avg3", line_color='#1469F4')
avg5 = go.Scatter(x=data['date'], y=data["avg5"], name="avg5", line_color='#089B5B')
avg10 = go.Scatter(x=data['date'], y=data["avg10"], name="avg10", line_color='#ff00ff')
avg20 = go.Scatter(x=data['date'], y=data["avg20"], name="avg20", line_color='#1469F4')
avg30 = go.Scatter(x=data['date'], y=data["avg30"], name="avg30", line_color='#FFA500')
avg20 = go.Scatter(x=data['date'], y=data["avg20"], name="avg20", line_color='#8F8203')
avg30 = go.Scatter(x=data['date'], y=data["avg30"], name="avg30", line_color='#000000')
#avg60 = go.Scatter(x=data['date'], y=data["avg60"], name="avg60", line_color='#008000')
candle_stick = go.Candlestick(x=data['date'], open=data['open'], high=data['high'], low=data['low'], close=data['close'], increasing_line_color='red', decreasing_line_color='blue')
@@ -164,8 +164,8 @@ class Simulation:
today = days[1]
# 데이터를 가지고 온다.
self.getCSV("./data/" + self.stock_code + "_" + last_day + ".csv", last_day, result)
self.getCSV("./data/" + self.stock_code + "_" + today + ".csv", today, result)
self.getCSV("./backup/" + self.stock_code + "_" + last_day + ".csv", last_day, result)
self.getCSV("./backup/" + self.stock_code + "_" + today + ".csv", today, result)
# 분석을 통해서 볼린저밴드 상/하단을 계산한다.
data = self.buySellChecker.analyze(result)
@@ -180,44 +180,41 @@ class Simulation:
if __name__ == "__main__":
"""
# to check bying
stock_codes = {
# 122630
"252670": [
('20210901', '20210902'),
('20210902', '20210903'),
('20210906', '20210907'),
('20210907', '20210908'),
('20210907', '20210908'),
('20210908', '20210909'),
('20210909', '20210910'),
('20210910', '20210913'),
('20210913', '20210914'),
('20210914', '20210915'),
('20210915', '20210916'),
('20210916', '20210917'),
('20210923', '20210924'),
('20210927', '20210928'),
('20210928', '20210929'),
('20210929', '20210930'),
('20210930', '20211001'),
('20211005', '20211006'),
('20211006', '20211007'),
('20211007', '20211008'),
('20211008', '20211009'),
('20211009', '20211011'),
('20211011', '20211012'),
('20211012', '20211013'),
('20211013', '20211014'),
('20211014', '20211015'),
('20220627', '20220628'),
('20220701', '20220704'),
],
}
"""
stock_codes = {
# 252670
# 122630
"122630": [
('20220617', '20220620'),
('20220620', '20220621'),
('20220621', '20220622'),
('20220622', '20220623'),
('20220623', '20220624'),
('20220624', '20220627'),
('20220627', '20220628'),
('20220628', '20220629'),
('20220629', '20220630'),
('20220630', '20220701'),
('20220701', '20220704'),
('20220704', '20220705'),
('20220705', '20220706'),
('20220706', '20220707'),
('20220707', '20220708')
],
}
for stock_code in stock_codes:
simulation = Simulation(stock_code)