AlanRex commited on
Commit
fe5a11e
·
verified ·
1 Parent(s): 80b5e37

Update model_predictor.py

Browse files
Files changed (1) hide show
  1. model_predictor.py +10 -1
model_predictor.py CHANGED
@@ -30,7 +30,12 @@ class XGBoostModel:
30
  'MACD_diff', # MACD - signal
31
  'dji_return_t-1', # 前一日道瓊指數報酬率
32
  'sox_return_t-1', # 前一日費半指數報酬率
33
- 'NEWS' # 新聞情緒分數
 
 
 
 
 
34
  ]
35
 
36
  # 【新增】輸出目標對應表
@@ -351,6 +356,10 @@ if __name__ == "__main__":
351
  'dji_return_t-1': [0.01],
352
  'sox_return_t-1': [0.015],
353
  'NEWS': [0.1]
 
 
 
 
354
  })
355
 
356
  print("測試模型預測器...")
 
30
  'MACD_diff', # MACD - signal
31
  'dji_return_t-1', # 前一日道瓊指數報酬率
32
  'sox_return_t-1', # 前一日費半指數報酬率
33
+ 'NEWS', # 新聞情緒分數
34
+ 'MACDvol',
35
+ 'RSI_14',
36
+ 'ADX',
37
+ 'volume_weighted_return'
38
+
39
  ]
40
 
41
  # 【新增】輸出目標對應表
 
356
  'dji_return_t-1': [0.01],
357
  'sox_return_t-1': [0.015],
358
  'NEWS': [0.1]
359
+ 'MACDvol': [0.015],
360
+ 'RSI_14': [0.015],
361
+ 'ADX': [0.015],
362
+ 'volume_weighted_return': [0.015],
363
  })
364
 
365
  print("測試模型預測器...")