xgb.train是XGBoost库中用于训练模型的函数。params参数是一个字典,用于设置XGBoost模型的超参数。常用的超参数包括:
- eta: 学习率,默认为0.3。
- max_depth: 树的最大深度,默认为6。
- subsample: 每棵树所用的子样本比例,默认为1。
- colsample_bytree: 每棵树所用的特征比例,默认为1。
- objective: 损失函数,默认为"reg:linear"。
- nthread: 使用的线程数,默认为当前系统的可用线程数。
- n_estimators: 树的数量,默认为100。
为了选择合适的参数,建议使用网格搜