当前位置:   article > 正文

机器学习(14)--网格搜索GridSearchCV,随机搜索RandomizedSearchCV_xgbclassifier gridsearchcv

xgbclassifier gridsearchcv

GridSearchCV

GridSearchCV的名字其实可以拆分为两部分,GridSearch和CV,即网格搜索和交叉验证。 
这两个概念都比较好理解,网格搜索,搜索的是参数,即在指定的参数范围内,按步长依次调整参数,利用调整的参数训练学习器,从所有的参数中找到在验证集上精度最高的参数,这其实是一个循环和比较的过程。 
GridSearchCV可以保证在指定的参数范围内找到精度最高的参数,但是这也是网格搜索的缺陷所在,它要求遍历所有可能参数的组合,在面对大数据集和多参数的情况下,非常耗时。这也是我通常不会使用GridSearchCV的原因,一般会采用后一种RandomizedSearchCV随机参数搜索的方法。

交叉验证的概念也很简单

· 将训练数据集划分为K份,K一般为10(我个人取3到5比较多) 
· 依次取其中一份为验证集,其余为训练集训练分类器,测试分类器在验证集上的精度 
· 取K次实验的平均精度为该分类器的平均精度

网格搜索就是利用交叉验证的形式比较每一个参数下训练器的精度的,但是交叉验证也要求大量的计算资源,加重了网格搜索的搜索时间

接下来以阿里IJCAI广告推荐数据集与XGBoostClassifier分类器为例,用代码的形式说明sklearn中GridSearchCV的使用方法

  1. import numpy as np
  2. import pandas as pd
  3. import xgboost as xgb
  4. from sklearn.grid_search import GridSearchCV
  5. #导入训练数据
  6. traindata = pd.read_csv("/home/pmqc01/LiangjunFeng/traindata_4_3_2.txt",sep = ',')
  7. traindata = traindata.set_index('instance_id')
  8. trainlabel = traindata['is_trade']
  9. del traindata['is_trade']
  10. print(traindata.shape,trainlabel.shape)
  11. #分类器使用 xgboost
  12. clf1 = xgb.XGBClassifier()
  13. #设定网格搜索的xgboost参数搜索范围,值搜索XGBoost的主要6个参数
  14. param_dist = {
  15. 'n_estimators':range(80,200,4),
  16. 'max_depth':range(2,15,1),
  17. 'learning_rate':np.linspace(0.01,2,20),
  18. 'subsample':np.linspace(0.7,0.9,20),
  19. 'colsample_bytree':np.linspace(0.5,0.98,10),
  20. 'min_child_weight':range(1,9,1)
  21. }
  22. #GridSearchCV参数说明,clf1设置训练的学习器
  23. #param_dist字典类型,放入参数搜索范围
  24. #scoring = 'neg_log_loss',精度评价方式设定为“neg_log_loss“
  25. #n_jobs = -1,使用所有的CPU进行训练,默认为1,使用1个CPU
  26. grid = GridSearchCV(clf1,param_dist,cv = 3,scoring = 'neg_log_loss',n_jobs = -1)
  27. #在训练集上训练
  28. grid.fit(traindata.values,np.ravel(trainlabel.values))
  29. #返回最优的训练器
  30. best_estimator = grid.best_estimator_
  31. print(best_estimator)
  32. #输出最优训练器的精度
  33. print(grid.best_score_)

这里关于网格搜索的几个参数再说明一下,评分参数“scoring“,需要根据实际的评价标准设定,阿里的IJCAI的标准是’neg_log_loss’,所以这里设定的是’neg_log_loss’,sklearn中备选的评价标准有一下:

 

这里写图片描述

 

在一些情况下,sklearn中没有现成的评价函数,sklearn是允许我们自己的定义的,但需要注意格式,接下来给个例子

  1. import numpy as np
  2. from sklearn.metrics import make_scorer
  3. def my_custom_loss_func(ground_truth, predictions):
  4. diff = np.abs(ground_truth - predictions).max()
  5. return np.log(1 + diff)
  6. #这里的greater_is_better参数决定了自定义的评价指标是越大越好还是越小越好
  7. loss = make_scorer(my_custom_loss_func, greater_is_better=False)
  8. score = make_scorer(my_custom_loss_func, greater_is_better=True)

定义好以后,再将其代入GridSearchCV函数就好

这里再贴一下常用的集成学习算法比较重要的需要调参的参数,供大家参考 
这里写图片描述

RandomizedSearchCV

RandomizedSearchCV的使用方法其实是和GridSearchCV一致的,但它以随机在参数空间中采样的方式代替了GridSearchCV对于参数的网格搜索,在对于有连续变量的参数时,RandomizedSearchCV会将其当作一个分布进行采样这是网格搜索做不到的,它的搜索能力取决于设定的n_iter参数,同样的给出代码

  1. import numpy as np
  2. import pandas as pd
  3. import xgboost as xgb
  4. from sklearn.grid_search import RandomizedSearchCV
  5. #导入训练数据
  6. traindata = pd.read_csv("/home/pmqc01/LiangjunFeng/traindata_4_3_2.txt",sep = ',')
  7. traindata = traindata.set_index('instance_id')
  8. trainlabel = traindata['is_trade']
  9. del traindata['is_trade']
  10. print(traindata.shape,trainlabel.shape)
  11. #分类器使用 xgboost
  12. clf1 = xgb.XGBClassifier()
  13. #设定搜索的xgboost参数搜索范围,值搜索XGBoost的主要6个参数
  14. param_dist = {
  15. 'n_estimators':range(80,200,4),
  16. 'max_depth':range(2,15,1),
  17. 'learning_rate':np.linspace(0.01,2,20),
  18. 'subsample':np.linspace(0.7,0.9,20),
  19. 'colsample_bytree':np.linspace(0.5,0.98,10),
  20. 'min_child_weight':range(1,9,1)
  21. }
  22. #RandomizedSearchCV参数说明,clf1设置训练的学习器
  23. #param_dist字典类型,放入参数搜索范围
  24. #scoring = 'neg_log_loss',精度评价方式设定为“neg_log_loss“
  25. #n_iter=300,训练300次,数值越大,获得的参数精度越大,但是搜索时间越长
  26. #n_jobs = -1,使用所有的CPU进行训练,默认为1,使用1个CPU
  27. grid = RandomizedSearchCV(clf1,param_dist,cv = 3,scoring = 'neg_log_loss',n_iter=300,n_jobs = -1)
  28. #在训练集上训练
  29. grid.fit(traindata.values,np.ravel(trainlabel.values))
  30. #返回最优的训练器
  31. best_estimator = grid.best_estimator_
  32. print(best_estimator)
  33. #输出最优训练器的精度
  34. print(grid.best_score_)
本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
  

闽ICP备14008679号