网格搜索参数及其常用代码

本文介绍了如何利用sklearn的GridSearchCV进行模型超参数的选择。内容涉及不同参数组合的性能表现,如0.01到100.0的步长与1到5的倍数的组合,并强调了该方法对计算机性能的要求。接着,文章概述了GridSearchCV的主要参数,包括cv_results_、best_estimator、best_score_和best_params_,并以随机森林模型为例说明了其应用。
摘要由CSDN通过智能技术生成

利用sklearn中的GridSearchCV对模型最优超参数进行选择,经常与交叉验证共用。参数选择表现如下:
0.01 0.1 1.0 10.0 100.0
1 (0.01, 1) (0.1, 1) (1, 1) (10, 1) (100, 1)
2 (0.01, 2) (0.1, 2) (1, 2) (10, 2) (100, 2)
3 (0.01, 3) (0.1, 3) (1, 3) (10, 3) (100, 3)
4 (0.01, 4) (0.1, 4) (1, 4) (10, 4) (100, 4)
5 (0.01, 5) (0.1, 5) (1, 5) (10, 5) (100, 5)
通过上述表格可以看出,网格搜索对电脑的性能要求是比较高的,因为会涉及到多次运算验证,所以代码一次性写好的要求比较高。接下来介绍网格搜索的主要参数及其函数。
首先,导入库,实例化模型,这里以随机森林为例。

import numpy as np
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import GridSearchCV
model = RandomForestClassifier()
param_grid = {'n_estimators': np.linspace(10, 100, 10).astype(int), 'max_depth': np.arange(4, 11)}
gs = GridSearchCV(model, param_grid=param_grid, scoring='accuracy', cv=5)
gs_res =
  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值