python对红酒数据集分别用留出法、交叉验证法进行划分的代码怎么写
时间: 2024-05-02 07:23:52 浏览: 58
您好,以下是Python对红酒数据集分别用留出法、交叉验证法进行划分的示例代码。
留出法:
```
from sklearn.model_selection import train_test_split
from sklearn.datasets import load_wine
# 加载红酒数据集
wine = load_wine()
# 划分数据集,将30%作为测试集
X_train, X_test, y_train, y_test = train_test_split(wine.data, wine.target, test_size=0.3, random_state=0)
```
交叉验证法:
```
from sklearn.model_selection import KFold
from sklearn.datasets import load_wine
# 加载红酒数据集
wine = load_wine()
# 5折交叉验证
kf = KFold(n_splits=5, shuffle=True, random_state=0)
for train_index, test_index in kf.split(wine.data):
X_train, X_test = wine.data[train_index], wine.data[test_index]
y_train, y_test = wine.target[train_index], wine.target[test_index]
```
希望这些代码能够帮助到您!
阅读全文