from sklearn.cross_valiation import train_test_split
时间: 2023-10-19 08:30:44 浏览: 71
使用sklearn的cross_val_score进行交叉验证实例
在sklearn库中,`train_test_split`函数用于将数据集划分为训练集和测试集。然而,从sklearn 0.22版本开始,`cross_valiation`模块已经被废弃,取而代之的是`model_selection`模块。
所以,正确的导入语句应该是:
```python
from sklearn.model_selection import train_test_split
```
这样你就可以使用`train_test_split`函数来进行数据集的划分了。
阅读全文