AlanRex commited on
Commit
be3e2e8
·
verified ·
1 Parent(s): 2a1032c

Update model_predictor.py

Browse files
Files changed (1) hide show
  1. model_predictor.py +8 -7
model_predictor.py CHANGED
@@ -20,6 +20,7 @@ class XGBoostModel:
20
  """
21
  self.model = None
22
  self.scaler = None
 
23
  self.feature_columns = [
24
  'close', # 前一日收盤價
25
  'return_t-1', # 前一日報酬率
@@ -30,12 +31,11 @@ class XGBoostModel:
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
  # 【新增】輸出目標對應表
@@ -47,7 +47,8 @@ class XGBoostModel:
47
  }
48
 
49
  print("XGBoost 模型預測器初始化完成")
50
- print("輸出格式:漲幅百分比 (1日, 5日, 10日, 20日)")
 
51
 
52
  def load_model(self, model_path):
53
  """
 
20
  """
21
  self.model = None
22
  self.scaler = None
23
+ # 【【修改點】】更新特徵欄位列表以包含新特徵
24
  self.feature_columns = [
25
  'close', # 前一日收盤價
26
  'return_t-1', # 前一日報酬率
 
31
  'MACD_diff', # MACD - signal
32
  'dji_return_t-1', # 前一日道瓊指數報酬率
33
  'sox_return_t-1', # 前一日費半指數報酬率
34
+ 'NEWS', # 新聞情緒分數
35
+ 'MACDvol', # 成交量MACD
36
+ 'RSI_14', # 14日RSI
37
+ 'ADX', # ADX趨勢指標
38
+ 'volume_weighted_return' # 成交量加權報酬率
 
39
  ]
40
 
41
  # 【新增】輸出目標對應表
 
47
  }
48
 
49
  print("XGBoost 模型預測器初始化完成")
50
+ print(f"輸出格式:漲幅百分比 (1日, 5日, 10日, 20日)")
51
+ print(f"預期特徵數量: {len(self.feature_columns)}")
52
 
53
  def load_model(self, model_path):
54
  """