贝叶斯优化调参

在机器学习项目中,一项十分重要的工作就是调整超参数,但有些算法的参数十分多,手动调参效率十分低下,而已存在的自动调参方法中网格搜索和随机搜索又十分缓慢,这里介绍一种贝叶斯调参法,为了知道该算法的实际应用,不会讲解原理,而是了解其实际应用。
这里采用随机森林算法作为讲解,实际应用时其它的算法调参时也可按照该模板进行调参。

#导入相应的库
from sklearn.datasets import make_classification
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import cross_val_score,KFold
from bayes_opt import BayesianOptimization
#生成一个简单的二分类数据集
x, y = make_classification(n_samples=1000,n_features=10,n_classes=2)
print(x.shape,y.shape)

>>>(1000, 10) (1000,)
#直接使用默认参数的随机森林进行分类
rf=RandomForestClassifier(random_state=0)
#划分出训练集和验证集
cv=KFold(n_splits=5,shuffle=True,random_state=0)
#计算交叉验证分数,这里的指标选择的是"roc_auo"
cross_val_score(rf,x,y,scoring="roc_auc",cv=cv).mean()

>>>0.9726355366078334

直接使用未调参的随机森林模型得出的结果为0.972,接下来对该模型进行贝叶斯调参。<

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值