python按比例随机切分数据
时间: 2023-08-11 16:23:45 浏览: 175
python按比例随机切分数据的实现
可以使用`sklearn.model_selection.train_test_split`函数来按比例随机切分数据。该函数可以将数据集随机划分成训练集和测试集,并且可以指定划分的比例。
示例代码如下:
```python
from sklearn.model_selection import train_test_split
X = # 特征矩阵
y = # 标签
# 将数据集按照 70% 的比例划分为训练集和测试集
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`为测试集。
阅读全文