XGboost-网格调参法(转载与测试)

一、原理

一篇好文:
https://www.jianshu.com/p/7467e616f227

二、网格调参
用xgboost既可以用来做二分类、多分类,也可以用来做回归预测数值,除了特征之外,影响模型的是如何调参了,一般是按一定的步骤、网格搜索最优参数,如下两篇文章一个是用来分类,一个是用来预测数值的案例,并且详细给出了调参的步骤和代码:
https://blog.csdn.net/han_xiaoyang/article/details/52665396 (用来分类XGBClassifier)
https://segmentfault.com/a/1190000014040317 (用来预测数字,XGBRegressor)

三、实践
参考以上的博客,用iris经典数据集进行多分类的预测(三个类别)。

  • 载入库
import pandas as pd
import numpy as np
import xgboost as xgb
from sklearn.model_selection import GridSearchCV
from xgboost.sklearn import XGBClassifier
from sklearn import metrics
import matplotlib.pyplot as plt
  • 数据集
from sklearn.datasets import load_iris
iris = load_iris()
X,y = iris.data,iris.target
  • 用Early Stop检查过拟合
model = XGBClassifier()
eval_set =[(X_test,y_test)]
model.fit
  • 0
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值