温馨提示×

Scikit-learn中怎么实现交叉验证

小亿
90
2024-05-10 17:20:57
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

在Scikit-learn中实现交叉验证可以使用cross_val_score函数。该函数可以帮助我们评估模型的性能,并且可以指定交叉验证的折数。

下面是一个示例代码:

from sklearn.model_selection import cross_val_score
from sklearn.datasets import load_iris
from sklearn.linear_model import LogisticRegression

# 加载数据集
iris = load_iris()
X, y = iris.data, iris.target

# 初始化模型
model = LogisticRegression()

# 进行交叉验证
scores = cross_val_score(model, X, y, cv=5)  # 5折交叉验证

print("交叉验证得分:", scores)
print("平均得分:", scores.mean())

在这个示例中,我们使用cross_val_score函数对一个逻辑回归模型进行了5折交叉验证,并输出了每一折的得分以及平均得分。您可以根据需要修改模型、数据集以及交叉验证的折数来实现不同的交叉验证方式。

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读:Scikit-learn中怎么绘制验证曲线

0