|
|
@@ -25,7 +25,7 @@ def parse_arguments():
|
|
|
parser.add_argument('--objective', type=str, default='reg:squarederror', help='XGBoost目标函数')
|
|
|
parser.add_argument('--learning_rate', type=float, default=0.1, help='学习率')
|
|
|
parser.add_argument('--max_depth', type=int, default=8, help='最大树深度')
|
|
|
- parser.add_argument('--min_child_weight', type=int, default=3, help='最小子权重')
|
|
|
+ parser.add_argument('--min_child_weight', type=float, default=3, help='最小子权重')
|
|
|
parser.add_argument('--gamma', type=float, default=2, help='gamma参数')
|
|
|
parser.add_argument('--subsample', type=float, default=0.85, help='子样本比例')
|
|
|
parser.add_argument('--colsample_bytree', type=float, default=0.75, help='每棵树的列采样率')
|
|
|
@@ -58,7 +58,7 @@ USE_HYPERPARAM_TUNING = False # 若 False 则直接使用 xgb.train
|
|
|
TARGET_COL = '美国RBOB汽油裂解'
|
|
|
TEST_PERIOD = 20
|
|
|
SEARCH_MODE = 'random' # 可选 'grid' / 'bayesian' / 'random'
|
|
|
-SHOW_PLOTS = False
|
|
|
+SHOW_PLOTS = True
|
|
|
ADJUST_FULL_PREDICTIONS = True
|
|
|
|
|
|
|
ziqidai11