Spaces:
Sleeping
Sleeping
Update model_predictor.py
Browse files- 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("測試模型預測器...")
|