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-12 03:14:55 +09:00
parent 95f68cab7b
commit a1d8bb10f3
4 changed files with 128 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
Simulation.py
View File

@@ -49,9 +49,10 @@ class Simulation (HTS):
'close': 'int',
'volume': 'int',
'avg3': 'float',
'avg5': 'float',
'avg10': 'float',
'avg20': 'float',
'avg6': 'float',
'avg9': 'float',
'avg12': 'float',
'avg24': 'float',
'avg30': 'float',
'avg60': 'float',
'fast_k': 'float',
@@ -69,7 +70,7 @@ class Simulation (HTS):
buy_line[i] = nan
buy_size.append(0)
else:
buy_colors.append("#00ced1")
buy_colors.append("#D87D0F")
buy_size.append(10 + (5 * buy_weight_line[i]))
sell_colors = []
@@ -78,7 +79,7 @@ class Simulation (HTS):
sell_colors.append("#ffffff")
sell_line[i] = nan
else:
sell_colors.append("#D87D0F")
sell_colors.append("#00ced1")
# 그래프를 설정한다.
buy_check = go.Scatter(x=data['date'], y=buy_line, mode='markers', name="buy", marker=dict(size=buy_size, color=buy_colors, line_width=0))
@@ -86,11 +87,11 @@ class Simulation (HTS):
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='#8F8203')
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='#000000')
avg60 = go.Scatter(x=data['date'], y=data["avg60"], name="avg60", line_color='#008000')
avg6 = go.Scatter(x=data['date'], y=data["avg6"], name="avg6", line_color='#089B5B')
avg9 = go.Scatter(x=data['date'], y=data["avg9"], name="avg9", line_color='#ff00ff')
avg12 = go.Scatter(x=data['date'], y=data["avg12"], name="avg12", line_color='#1469F4')
avg24 = go.Scatter(x=data['date'], y=data["avg24"], name="avg24", line_color='#000000')
avg30 = go.Scatter(x=data['date'], y=data["avg30"], name="avg30", 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')
volume_line = go.Scatter(x=data['date'], y=data["volume"], mode='lines', name='volume')
@@ -106,7 +107,7 @@ class Simulation (HTS):
rsi_line = go.Scatter(x=data['date'], y=data["rsi"], mode='lines', name='rsi')
rsis_line = go.Scatter(x=data['date'], y=data["rsis"], mode='lines', name='rsis')
candle_data = [candle_stick, upper, lower, avg3, avg5, avg10, avg20, avg30, avg60, buy_check, sell_check]
candle_data = [candle_stick, upper, lower, avg3, avg6, avg9, avg12, avg24, avg30, buy_check, sell_check]
volume_data = [volume_line]
macd_data = [macd_line, macd_s_line, macd_o_line]
stochastic_data = [slow_k_line, slow_d_line]
@@ -180,7 +181,7 @@ if __name__ == "__main__":
# to check bying
stock_codes = {
"252670": ['20220801', '20220802', '20220803', '20220804', '20220805', '20220808', '20220809', '20220810', '20220811'],
#"122630": ['20220810'],
"122630": ['20220801', '20220802', '20220803', '20220804', '20220805', '20220808', '20220809', '20220810', '20220811'],
}
method = "rule" # "rule", "ml", "answer"