溫馨提示×

Scikit-learn中怎么實現交叉驗證

小億
92
2024-05-10 17:20:57
欄目: 編程語言

在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折交叉驗證,并輸出了每一折的得分以及平均得分。您可以根據需要修改模型、數據集以及交叉驗證的折數來實現不同的交叉驗證方式。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女