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

init

Browse Source
This commit is contained in:
dsyoon
2023-12-21 01:27:52 +09:00
parent 1f0fccd7be
commit aba1314468
2 changed files with 116 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

121
HTS_etf.py
View File

@@ -143,6 +143,8 @@ class HTS_etf(HTS):
avg120 = [item[0] for item in avg120_list]
avg240_list = close_df.rolling(window=240).mean().fillna(close[0]).values.tolist()
avg240 = [item[0] for item in avg240_list]
avg420_list = close_df.rolling(window=420).mean().fillna(close[0]).values.tolist()
avg420 = [item[0] for item in avg420_list]
avg480_list = close_df.rolling(window=480).mean().fillna(close[0]).values.tolist()
avg480 = [item[0] for item in avg480_list]
avg1500_list = close_df.rolling(window=1500).mean().fillna(close[0]).values.tolist()
@@ -214,11 +216,11 @@ class HTS_etf(HTS):
if "volume_up" in result and "volume_updown_diff" in result:
for i in range(len(open)):
STOCK.append({'volume': volume[i], 'volume_down': volume_down[i], 'volume_up': volume_up[i], 'volume_updown_diff': volume_updown_diff[i], 'close': close[i], 'open': open[i], 'high': high[i], 'low': low[i],
'avg5': avg5[i], 'avg20': avg20[i], 'avg60': avg60[i], 'avg120': avg120[i], 'avg240': avg240[i], 'avg480': avg480[i], 'avg1500': avg1500[i]})
'avg5': avg5[i], 'avg20': avg20[i], 'avg60': avg60[i], 'avg120': avg120[i], 'avg240': avg240[i], 'avg420': avg420[i], 'avg480': avg480[i], 'avg1500': avg1500[i]})
else:
for i in range(len(open)):
STOCK.append({'volume': volume[i], 'close': close[i], 'open': open[i], 'high': high[i], 'low': low[i],
'avg5': avg5[i], 'avg20': avg20[i], 'avg60': avg60[i], 'avg120': avg120[i], 'avg240': avg240[i], 'avg480': avg480[i], 'avg1500': avg1500[i]})
'avg5': avg5[i], 'avg20': avg20[i], 'avg60': avg60[i], 'avg120': avg120[i], 'avg240': avg240[i], 'avg420': avg420[i], 'avg480': avg480[i], 'avg1500': avg1500[i]})
# stochastic
stochastic_df = self.stochastic.apply(STOCK, n=30, m=5, t=5)
@@ -254,7 +256,7 @@ class HTS_etf(HTS):
"open": open, "high": high, "low": low, "close": close, "volume": volume, "volume_down": volume_down,
"volume_up": volume_up, "volume_updown_diff": volume_updown_diff,
"trend": trend, "trend_avg": trend_avg,
"avg5": avg5, "avg20": avg20, "avg60": avg60, "avg120": avg120, "avg240": avg240, "avg480": avg480,
"avg5": avg5, "avg20": avg20, "avg60": avg60, "avg120": avg120, "avg240": avg240, "avg420": avg420, "avg480": avg480,
"avg1500": avg1500,
"disparity_avg5": disparity_avg5, "disparity_avg20": disparity_avg20,
"disparity_avg30": disparity_avg30, "disparity_avg60": disparity_avg60,
@@ -273,7 +275,7 @@ class HTS_etf(HTS):
"ymd": point_temp,
"open": open, "high": high, "low": low, "close": close, "volume": volume,
"trend": trend, "trend_avg": trend_avg,
"avg5": avg5, "avg20": avg20, "avg60": avg60, "avg120": avg120, "avg240": avg240, "avg480": avg480,
"avg5": avg5, "avg20": avg20, "avg60": avg60, "avg120": avg120, "avg240": avg240, "avg420": avg420, "avg480": avg480,
"avg1500": avg1500,
"disparity_avg5": disparity_avg5, "disparity_avg20": disparity_avg20,
"disparity_avg30": disparity_avg30, "disparity_avg60": disparity_avg60,
@@ -422,26 +424,46 @@ class HTS_etf(HTS):
disparity = {
'avg': {},
"limit_top_1": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_bottom_1": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_top_3": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_bottom_3": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_top_5": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_bottom_5": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_top_10": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_bottom_10": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_top_20": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_bottom_20": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_top_20": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_bottom_20": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_top_20": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_bottom_20": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_top_30": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_bottom_30": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_top_40": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_bottom_40": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_top_99": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None},
"limit_bottom_99": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None, "avg1500": None}
"limit_top_1": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None,
"avg1500": None},
"limit_bottom_1": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None,
"avg480": None, "avg1500": None},
"limit_top_3": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None,
"avg1500": None},
"limit_bottom_3": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None,
"avg480": None, "avg1500": None},
"limit_top_5": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None,
"avg1500": None},
"limit_bottom_5": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None,
"avg480": None, "avg1500": None},
"limit_top_10": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None,
"avg1500": None},
"limit_bottom_10": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None,
"avg480": None, "avg1500": None},
"limit_top_20": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None,
"avg1500": None},
"limit_bottom_20": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None,
"avg480": None, "avg1500": None},
"limit_top_20": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None,
"avg1500": None},
"limit_bottom_20": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None,
"avg480": None, "avg1500": None},
"limit_top_20": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None,
"avg1500": None},
"limit_bottom_20": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None,
"avg480": None, "avg1500": None},
"limit_top_30": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None,
"avg1500": None},
"limit_bottom_30": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None,
"avg480": None, "avg1500": None},
"limit_top_98": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None,
"avg1500": None},
"limit_bottom_98": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None,
"avg480": None, "avg1500": None},
"limit_top_99": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None, "avg480": None,
"avg1500": None},
"limit_bottom_99": {"avg5": None, "avg20": None, "avg60": None, "avg120": None, "avg240": None,
"avg480": None, "avg1500": None}
}
disparity['avg'] = avg
disparity_1500 = sorted(list(set(avg['avg1500'])), reverse=True)
@@ -452,39 +474,28 @@ class HTS_etf(HTS):
disparity_20 = sorted(list(set(avg['avg20'])), reverse=True)
disparity_5 = sorted(list(set(avg['avg5'])), reverse=True)
poses = [1, 3, 5, 10, 20, 30, 40, 99]
poses = [1, 3, 5, 10, 20, 30, 98, 99]
for pos in poses:
loc = pos
if pos == 99:
pos = int(len(disparity_20) * 0.03)
disparity['limit_top_99']['avg1500'] = disparity_1500[pos - 1]
disparity['limit_bottom_99']['avg1500'] = disparity_1500[len(disparity_1500) - pos - 1]
disparity['limit_top_99']['avg480'] = disparity_480[pos]
disparity['limit_bottom_99']['avg480'] = disparity_480[len(disparity_480) - pos - 1]
disparity['limit_top_99']['avg240'] = disparity_240[pos]
disparity['limit_bottom_99']['avg240'] = disparity_240[len(disparity_240) - pos - 1]
disparity['limit_top_99']['avg120'] = disparity_120[pos]
disparity['limit_bottom_99']['avg120'] = disparity_120[len(disparity_120) - pos - 1]
disparity['limit_top_99']['avg60'] = disparity_60[pos]
disparity['limit_bottom_99']['avg60'] = disparity_60[len(disparity_60) - pos - 1]
disparity['limit_top_99']['avg20'] = disparity_20[pos]
disparity['limit_bottom_99']['avg20'] = disparity_20[len(disparity_20) - pos - 1]
disparity['limit_top_99']['avg5'] = disparity_5[pos]
disparity['limit_bottom_99']['avg5'] = disparity_5[len(disparity_5) - pos - 1]
else:
disparity['limit_top_' + str(pos)]['avg1500'] = disparity_1500[pos - 1]
disparity['limit_bottom_' + str(pos)]['avg1500'] = disparity_1500[len(disparity_1500) - pos - 1]
disparity['limit_top_' + str(pos)]['avg480'] = disparity_480[pos]
disparity['limit_bottom_' + str(pos)]['avg480'] = disparity_480[len(disparity_480) - pos - 1]
disparity['limit_top_' + str(pos)]['avg240'] = disparity_240[pos]
disparity['limit_bottom_' + str(pos)]['avg240'] = disparity_240[len(disparity_240) - pos - 1]
disparity['limit_top_' + str(pos)]['avg120'] = disparity_120[pos]
disparity['limit_bottom_' + str(pos)]['avg120'] = disparity_120[len(disparity_120) - pos - 1]
disparity['limit_top_' + str(pos)]['avg60'] = disparity_60[pos]
disparity['limit_bottom_' + str(pos)]['avg60'] = disparity_60[len(disparity_60) - pos - 1]
disparity['limit_top_' + str(pos)]['avg20'] = disparity_20[pos]
disparity['limit_bottom_' + str(pos)]['avg20'] = disparity_20[len(disparity_20) - pos - 1]
disparity['limit_top_' + str(pos)]['avg5'] = disparity_5[pos]
disparity['limit_bottom_' + str(pos)]['avg5'] = disparity_5[len(disparity_5) - pos - 1]
loc = int(len(disparity_20) * 0.03)
elif pos == 98:
loc = int(len(disparity_20) * 0.05)
disparity['limit_top_' + str(pos)]['avg1500'] = disparity_1500[loc - 1]
disparity['limit_bottom_' + str(pos)]['avg1500'] = disparity_1500[len(disparity_1500) - loc - 1]
disparity['limit_top_' + str(pos)]['avg480'] = disparity_480[pos]
disparity['limit_bottom_' + str(pos)]['avg480'] = disparity_480[len(disparity_480) - loc - 1]
disparity['limit_top_' + str(pos)]['avg240'] = disparity_240[pos]
disparity['limit_bottom_' + str(pos)]['avg240'] = disparity_240[len(disparity_240) - loc - 1]
disparity['limit_top_' + str(pos)]['avg120'] = disparity_120[pos]
disparity['limit_bottom_' + str(pos)]['avg120'] = disparity_120[len(disparity_120) - loc - 1]
disparity['limit_top_' + str(pos)]['avg60'] = disparity_60[pos]
disparity['limit_bottom_' + str(pos)]['avg60'] = disparity_60[len(disparity_60) - loc - 1]
disparity['limit_top_' + str(pos)]['avg20'] = disparity_20[pos]
disparity['limit_bottom_' + str(pos)]['avg20'] = disparity_20[len(disparity_20) - loc - 1]
disparity['limit_top_' + str(pos)]['avg5'] = disparity_5[pos]
disparity['limit_bottom_' + str(pos)]['avg5'] = disparity_5[len(disparity_5) - loc - 1]
return disparity