划分图片形式的数据集,使得随机划分的训练集和测试集的比例为9:1
时间: 2024-05-03 11:19:48 浏览: 91
给定比例随机划分训练集、验证集和测试集
可以使用sklearn库中的train_test_split函数来划分数据集。代码示例如下:
```python
from sklearn.model_selection import train_test_split
X = # 图片数据
y = # 图片标签
# 划分训练集和测试集,test_size为测试集所占比例
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.1, random_state=42)
```
其中,X和y分别表示图片数据和标签,test_size为测试集所占比例,random_state为随机种子,保证每次划分的结果相同。上述代码中,训练集占比为0.9,测试集占比为0.1,符合题目要求的9:1比例。
阅读全文