add more strategies

This commit is contained in:
jackyu66git
2025-05-09 21:10:45 +08:00
parent 65a85823b9
commit 2440e896ea
17 changed files with 887 additions and 271 deletions
+18 -9
View File
@@ -16,6 +16,8 @@ class ChanKLU:
self.ma5 = 0
self.ma10 = 0
self.ma30 = 0
self.ma50 = 0
self.ma200 = 0
self.ma250 = 0
self.rsi = 0
self.volume_ratio = 0
@@ -23,15 +25,20 @@ class ChanKLU:
self.idx = idx
self.index = idx
def set_indicators(self, item):
self.macd = float(item['macd']) if item['macd'] else 0
self.signal = float(item['macdsignal']) if item['macdsignal'] else 0
self.macdhist = float(item['macdhist']) if item['macdhist'] else 0
self.ma5 = float(item['ma5']) if item['ma5'] else 0
self.ma10 = float(item['ma10']) if item['ma10'] else 0
self.ma30 = float(item['ma30']) if item['ma30'] else 0
self.ma250 = float(item['ma250']) if item['ma250'] else 0
self.rsi = float(item['rsi']) if item['rsi'] else 0
self.volume_ratio = float(item['volume_ratio']) if item['volume_ratio'] else 0
self.macd = float(item['macd']) if 'macd' in item and item['macd'] else 0
self.signal = float(item['macdsignal']) if 'macdsignal' in item and item['macdsignal'] else 0
self.macdhist = float(item['macdhist']) if 'macdhist' in item and item['macdhist'] else 0
self.ma5 = float(item['ma5']) if 'ma5' in item and item['ma5'] else 0
self.ma10 = float(item['ma10']) if 'ma10' in item and item['ma10'] else 0
self.ma30 = float(item['ma30']) if 'ma30' in item and item['ma30'] else 0
# 安全检查 ma250、ma50 和 ma200
self.ma250 = float(item['ma250']) if 'ma250' in item and item['ma250'] else 0
self.ma50 = float(item['ma50']) if 'ma50' in item and item['ma50'] else 0
self.ma200 = float(item['ma200']) if 'ma200' in item and item['ma200'] else 0
self.rsi = float(item['rsi']) if 'rsi' in item and item['rsi'] else 0
self.volume_ratio = float(item['volume_ratio']) if 'volume_ratio' in item and item['volume_ratio'] else 0
def get_feature_data(self):
features = dict()
features['klu_close'] = self.close
@@ -46,6 +53,8 @@ class ChanKLU:
features['klu_ma5'] = self.ma5
features['klu_ma10'] = self.ma10
features['klu_ma30'] = self.ma30
features['klu_ma50'] = self.ma50
features['klu_ma200'] = self.ma200
features['klu_ma250'] = self.ma250
features['klu_rsi'] = self.rsi
features['klu_volume_ratio'] = self.volume_ratio