K折交叉验证

在这里插入图片描述
将数据分成训练集,验证集,测试集
其中将训练集+验证集的K折交叉验证,就是将数据分成K等分,其中1份作为验证集,剩余的k-1作为训练集,对模型进行拟合,得到的平均值,就是这个模型的score,如下就是简单的示例,其中 train_table, train_labels是已经处理好的数据集

 	from sklearn.model_selection import cross_val_score
    #x_train, x_test, y_train, y_test = train_test_split(train_table, train_labels, test_size=0.2, random_state=0)
    from sklearn.ensemble import RandomForestClassifier
    clf = RandomForestClassifier(n_estimators=100, max_depth=None, min_samples_split=2, random_state=0)
    scores = cross_val_score(clf, train_table, train_labels, cv=5)
    print("Accuracy: %0.2f (+/- %0.2f)" % (scores.mean(), scores.std() * 2))

    Accuracy: 0.98 (+/- 0.02)

方法2:

 	 from sklearn.model_selection import ShuffleSplit
     n_samples = iris.data.shape[0]
     cv = ShuffleSplit(n_splits=5, test_size=0.3, random_state=0)
     cross_val_score(clf, iris.data, iris.target, cv=cv)  
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值