在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折交叉验证,并输出了每一折的得分以及平均得分。您可以根据需要修改模型、数据集以及交叉验证的折数来实现不同的交叉验证方式。