python训练集测试集划分
时间: 2023-09-04 08:13:19 浏览: 55
python 划分数据集为训练集和测试集的方法
在Python中,可以使用scikit-learn库中的train_test_split()函数来进行训练集测试集划分。该函数将数据集随机分成训练集和测试集,可以控制分割比例和随机种子。
下面是一个示例代码:
```python
from sklearn.model_selection import train_test_split
# X为特征数据,y为标签数据
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
```
其中,X为特征数据,y为标签数据,test_size为测试集所占的比例(这里为20%),random_state为随机种子,可以固定随机结果。该函数返回四个数组,分别为训练集特征数据、测试集特征数据、训练集标签数据、测试集标签数据。
阅读全文