模型的交叉验证与混淆矩阵(二分类)

一、交叉验证

交叉验证:就是在训练集中再抽出一部分当作测试,比如总数据本分成10分,训练集7份,测试集3份,在7份训练集中在抽出份当作测试,也就是说拿前6份训练时,把第7份当做测试。训练集有7份就要训练7次,这7份数据轮流当测试集,结果的准确性是这7次训练的平均值。

分类模型的交叉验证

from sklearn import datasets
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import train_test_split

cancer = datasets.load_breast_cancer()
cancer_x = cancer.data
cancer_y = cancer.target

set(cancer_y)#y是二分类数据
#返回值
{
   0, 1}

#将数据集分为训练集和测试集
x_train1, x_test1, y_train1, y_test1 = train_test_split(cancer_x, cancer_y, test_size = 0.3) 

#将模型实例化
rf = RandomForestClassifier(criterion = 'entropy', max_depth = 3, n_estimators = 5, random_state = 300)

#训练模型
rf = rf.fit(x_train1, y_train1)

1、随机子抽样验证(Hold-Out Method)

from sklearn.model_selection import cross_val_score

#cv=3表示交叉验证做3次
#scoring='accuracy'适用于分类模型
cross_val_score(rf,x_train1,y_train1,cv=3
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值