参考某篇QSAR的sci论文设置
根据上图,我设置我的XGBoost模型:
# 定义要搜索的超参数的候选值
param_grid = {'model__learning_rate': [0.1, 0.01, 0.001], # 调整学习率'model__n_estimators': [50, 100, 200, 300,400,500], # 调整树的数量'model__max_depth': [3,4,5,6,7,8,9], # 调整树的最大深度'model__subsample': [0.6,0.7,0.8, 0.9, 1.0], # 调整子样本的比例'model__colsample_bytree': [0.6,0.7,0.8, 0.9, 1.0] # 每棵树使用的特征比例
}