Spaces:
Sleeping
Sleeping
Update model_predictor.py
Browse files- 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 |
"""
|