python随机划分训练集和测试集
时间: 2023-09-05 13:08:42 浏览: 93
Python 实现训练集、测试集随机划分
在Python中,可以使用scikit-learn库中的train_test_split函数来随机划分训练集和测试集。该函数的语法如下:
```python
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
```
其中,X和y分别是特征矩阵和目标向量,test_size表示测试集所占比例,random_state表示随机种子,用于保证每次划分结果一致。该函数会返回四个变量,即训练集特征矩阵、测试集特征矩阵、训练集目标向量和测试集目标向量。可以根据需要进行相应的处理和训练。
阅读全文