pytorch-Train-Val-Test划分(下)
pytorch-Train-Val-Test划分(下)
2019-11-17阅读2860
上半节讲到一个数据集正常需要划分为train、validation和test三个数据集,那么具体到代码中是如何实现的?
需要注意的是正常代码提供的是两种划分,即train和test两部分,我们在划分train的数据集中再划分为两部分,即可设定val数据集。
代码如下
print('train:'...
转载
2020-04-08 11:56:14 ·
2853 阅读 ·
0 评论