已知X,y,随机划分训练集测试集的代码
时间: 2024-05-12 10:19:10 浏览: 80
划分训练集和测试集 示例代码 python
以下是Python中随机划分训练集和测试集的代码:
```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)
```
其中,`test_size`表示测试集所占比例,`random_state`表示随机种子,用于保证每次划分的结果一致。函数返回4个数组,分别为训练集和测试集的特征矩阵和标签。
阅读全文