python 切分训练集
时间: 2023-10-29 21:00:18 浏览: 111
python 划分数据集为训练集和测试集的方法
您可以使用Python中的sklearn库来切分训练集。sklearn库中的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参数用于指定测试集的比例,这里的0.2表示测试集占总数据的20%。random_state参数用于设置随机种子,以确保每次运行代码时得到相同的切分结果。
切分完成后,X_train和y_train分别是训练集的特征和标签,X_test和y_test分别是测试集的特征和标签。您可以根据需要进行模型训练和评估。
阅读全文