划分训练集和测试集python代码实现
时间: 2023-08-22 08:03:45 浏览: 121
在Python中,我们可以使用sklearn库中的train_test_split函数来划分训练集和测试集。具体的实现代码如下所示:
```python
from sklearn.model_selection import train_test_split
# X为特征数据,y为标签数据,test_size为测试集占比
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
```
其中,X为特征数据,y为标签数据,test_size表示测试集占总数据集的比例,random_state为随机数种子,用于保证每次划分的结果一致。划分后,我们可以使用X_train和y_train来训练模型,使用X_test和y_test来测试模型性能。
阅读全文