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
2024-01-25 10:26:56 +09:00
parent 4592897fe0
commit 0bbf1ba43b
1 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
stock/analysis/AnalyzerSqlite.py
View File

@@ -726,16 +726,20 @@ class AnalyzerSqlite:
self.macd.analyze(stock)
close_list = [price['close'] for price in stock['PRICE']]
# 트렌드: 실시간 정상 작동 안됨
size = int(len(close_list) / 20)
pos = round(size / 2)
close_temp = close_list + [close_list[-1]] * pos
decomposition_results = seasonal_decompose(close_temp, model='multiplicative', period=size)
trend = decomposition_results.trend[:-pos]
trend_df = pd.DataFrame(trend).fillna(close_list[0])
trend_list = trend_df.values.tolist()
for i, price in enumerate(stock['PRICE']):
price['trend'] = trend_list[i][0]
if 200 < len(close_list):
# 트렌드: 실시간 정상 작동 안됨
size = int(len(close_list) / 20)
pos = round(size / 2)
close_temp = close_list + [close_list[-1]] * pos
decomposition_results = seasonal_decompose(close_temp, model='multiplicative', period=size)
trend = decomposition_results.trend[:-pos]
trend_df = pd.DataFrame(trend).fillna(close_list[0])
trend_list = trend_df.values.tolist()
for i, price in enumerate(stock['PRICE']):
price['trend'] = trend_list[i][0]
else:
for i, price in enumerate(stock['PRICE']):
price['trend'] = stock['PRICE'][i]['avg120']
sorted_stock = sorted(stock["PRICE"], key=lambda x: x['ymd'], reverse=True)
for price in sorted_stock: