AIEnhancedStrategy

This commit is contained in:
zhangkun9038@dingtalk.com 2025-05-01 19:02:53 +08:00
parent 028a616f36
commit dac9005052

View File

@ -50,12 +50,21 @@
"include_volume_data": true,
"include_bidirectional_training_data": true
},
"data_split_parameters": {
"test_size": 0.2,
"shuffle": true,
"random_state": 42
},
"model_training_parameters": {
"model_type": "LightGBMRegressor",
"n_estimators": 100,
"learning_rate": 0.1,
"max_depth": 10,
"early_stopping_rounds": 10
"n_estimators": 200,
"learning_rate": 0.05,
"max_depth": 5,
"subsample": 0.8,
"colsample_bytree": 0.8,
"objective": "reg:squarederror",
"eval_metric": "rmse",
"early_stopping_rounds": 50,
"verbose": 0
}
},
"dry_run": true,