使用sklearn进行调参

竟整这些花里胡哨的,直接嵌套for完了。手动狗头。
官方文档

from sklearn.model_selection import GridSearchCV

关键参数

estimator:模型
param_grid:需要测试的参数,一个字典,字典中每个元素是超参数,示例:

param_grid= {'criterion':['gini'],'max_depth':[30,50,60,100],'min_samples_leaf':[2,3,5,10],'min_impurity_decrease':[0.1,0.2,0.5]}

注意:字典里面的值必须是李彪,即使单元素也要求是单元素列表!
也可以是一个列表,列表中每个元素是字典,会逐个字典去调参测试。
cv:k折交叉验证
scoring:评价标准,所有的评价标准参考:metrics
n_jos:几核cpu

关键属性

best_params_:最佳超参数
best_score_:最佳得分

关键方法

.fit(x_train,y_train):训练

使用示例


clf = GradientBoostingClassifier()
grid = GridSearchCV(clf, params, cv=10, scoring="f1")
grid.fit(X, y)
print(grid.best_params_)
  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值