Initial commit

2025-04-22 10:04:30 +08:00
commit 626d3a6f26
43 changed files with 8607 additions and 0 deletions
@@ -0,0 +1,40 @@
+class ChanKLU:
+    def __init__(self, time, open, high, low, close, volume):
+        # _time, _close, _open, _high, _low, _extra_info={}
+        self.kl_type = None
+        self.time = time
+        self.close = close
+        self.open = open
+        self.high = high
+        self.low = low
+        self.volume = volume
+        self.idx = 0
+        self.index = 0
+        self.macd = 0
+        self.signal = 0
+        self.macdhist = 0
+        self.ma5 = 0
+        self.ma10 = 0
+        self.ma30 = 0
+        self.ma250 = 0
+        self.rsi = 0
+    def set_idx(self, idx):
+        self.idx = idx
+        self.index = idx
+    def get_feature_data(self):
+        features = dict()
+        features['klu_close'] = self.close
+        features['klu_open'] = self.open
+        features['klu_high'] = self.high
+        features['klu_low'] = self.low
+        features['klu_volume'] = self.volume
+        features['klu_index'] = self.index
+        features['klu_macd'] = self.macd
+        features['klu_signal'] = self.signal
+        features['klu_macdhist'] = self.macdhist
+        features['klu_ma5'] = self.ma5
+        features['klu_ma10'] = self.ma10
+        features['klu_ma30'] = self.ma30
+        features['klu_ma250'] = self.ma250
+        features['klu_rsi'] = self.rsi
+        return features