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-06-24 21:25:57 +09:00
parent 0d523678a6
commit 0b4a87b334
5 changed files with 176 additions and 281 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
hts/Simulation.py
View File

@@ -74,7 +74,11 @@ class Simulation:
data['low'] = pd.to_numeric(data['low'])
data['close'] = pd.to_numeric(data['close'])
data['volume'] = pd.to_numeric(data['volume'])
data['avg2'] = pd.to_numeric(data['avg2'])
data['avg5'] = pd.to_numeric(data['avg5'])
data['avg10'] = pd.to_numeric(data['avg10'])
data['avg30'] = pd.to_numeric(data['avg30'])
data['avg60'] = pd.to_numeric(data['avg60'])
data["fast_k"] = pd.to_numeric(data['fast_k'])
data["slow_k"] = pd.to_numeric(data['slow_k'])
data["slow_d"] = pd.to_numeric(data['slow_d'])
@@ -101,6 +105,7 @@ 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))
bolinger_upper = go.Scatter(x=data['date'], y=data["upper"], name="upper", line_color='#8B4513')
bolinger_lower = go.Scatter(x=data['date'], y=data["lower"], name="lower", line_color='#8B4513')
avg2 = go.Scatter(x=data['date'], y=data["avg2"], name="avg2", line_color='#800080')
avg5 = go.Scatter(x=data['date'], y=data["avg5"], name="avg5", line_color='#800080')
avg10 = go.Scatter(x=data['date'], y=data["avg10"], name="avg10", line_color='#ff00ff')
avg30 = go.Scatter(x=data['date'], y=data["avg30"], name="avg30", line_color='#00ffff')
@@ -115,9 +120,9 @@ class Simulation:
rsis_line = go.Scatter(x=data['date'], y=data["rsis"], mode='lines', name='rsis')
#candle_data = [candle_stick, bolinger_upper, bolinger_lower, buy_check, sell_check, avg1, avg2, avg5, avg10, avg20, avg30, avg40, avg50, avg60]
candle_data = [candle_stick, bolinger_upper, bolinger_lower, avg5, avg10, avg30, avg60, buy_check, sell_check]
candle_data = [candle_stick, bolinger_upper, bolinger_lower, avg2, avg5, avg10, avg30, avg60, buy_check, sell_check]
volume_data = [volume_line]
stochastic_data = [slow_k_line, slow_d_line]
stochastic_data = [fast_k_line, slow_k_line, slow_d_line]
rsi_data = [rsi_line, rsis_line]
# 그래프를 그린다.
@@ -179,12 +184,15 @@ if __name__ == "__main__":
PROJECT_HOME = os.path.join(os.path.dirname(os.path.join(os.path.dirname(__file__))))
RESOURCE_DIR = PROJECT_HOME + "/resources/analysis/"+today.strftime("%Y%m%d")
stock_codes = ["252670", "122630"]
given_days = ['20210901', '20211020', '20220128', '20220520']
simulation = Simulation(stock_codes[0])
stock_codes = {
"252670": ['20220520', '20220128', '20220121', '20220120', '20211026'],
"122630": ['20211026', '20211025', '20211022', '20211021', '20211020']
}
given_days = sorted(given_days, reverse=True)
for given_day in given_days:
simulation.simulate(given_day)
for stock_code in stock_codes:
simulation = Simulation(stock_code)
for given_day in stock_codes[stock_code]:
simulation.simulate(given_day)
print ("done...")