欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

使用sklearn cross_val_score进行交叉验证

程序员文章站 2022-07-14 11:43:56
...
from sklearn.model_selection import  cross_val_score
score = cross_val_score(model, train_data.values, train_y, scoring="neg_mean_squared_error", cv = kf)

入参说明

estimator模型
X:特征 n(样本数量)行m(特征数量)列
y:预测数据
cv:交叉验证生成器或可迭代次数
n_jobs:工作的CPU数量
scoring:评价方法
使用sklearn cross_val_score进行交叉验证
主要就是上面这些,剩下的以后补充。

相关标签: 机器学习