求pytorch大神指点如何将自己的数据集按8:2划分为训练
pytorch吧
全部回复
仅看楼主
level 1
龙吟一声天下惊 楼主
求pytorch大神指点如何将自己的数据集按8:2划分为训练集和测试集
2018年05月03日 15点05分 1
level 1
🌚☜戳你脑门儿
随机数选。然后按比例
2018年05月05日 13点05分 2
龙吟一声天下惊
大神，那我分出数据集之后怎么在模型评估那里按照train和val来进行测试呢？
2018年05月05日 13点05分
龙吟一声天下惊
我看网上的都是用trainloader（其中trainloader＝torch.utils.data.DataLoader()来做的）这个来弄
2018年05月05日 13点05分
🌚☜戳你脑门儿
@龙吟一声天下惊 图的话 imageloader 也行 dataloader就是个自动迭代器。输数据用的
2018年05月06日 01点05分
龙吟一声天下惊
回复 🌚☜戳你脑门儿 :大神，能请你帮我写一份能将测试集分为train和val，然后网络用lenet5，然后能训练[乖]可以给你报酬的
2018年05月06日 04点05分
level 1
glove😜
trainset, testset = train_test_split(img_data, test_size = 0.2)即可，train_test_split是sklearn.cross_validation的一个可调用分配数据集函数。
2018年06月16日 01点06分 3
level 1
葛庄脾气暴公社
from sklearn.cross_validation import train_test_split
X_train, X_test, y_train, y_test = train_test_split( X, y, test_size=0.2)
2018年10月05日 02点10分 4
1