Pytorch加载数据是如何区分训练集和测试集

2025-04-15 20:40:19

1、1.Pytorch加载数据是如何区分训练集和测试集？

2、2.数据加载的语句如下图所示

3、3.可以发现训练集和测试集的数据集的加载使用的是通用的加载函数

4、4.该函数的第一个参数是数据集的路径

5、5.第二个参数为是否参与训练过程，不参与则为测试集数据，参与则为训练集数据

6、6.因此将第二个参数设置为True时则为训练集的数据集

声明：本网站引用、摘录或转载内容仅供网站访问者交流或参考，不代表本站立场，如存在版权或非法内容，请联系站长删除，联系邮箱：site.kefu@qq.com。