如何在C ++ pyTorch中创建自己的数据集

时间:2019-01-22 01:47:35

标签: c++ pytorch

我正在尝试使用pyTorch C ++进行一些机器学习。我在.csv文件中有自己的数据集,加载它们后收到例如。具有N个样本和M个特征的[N,M] 2D std :: vector和具有N个标签的[N,] 1D std :: vector。

我的问题是,我应该怎么做才能创建适当的数据集以使用torch :: data :: make_data_loader方法(或其他方法)并学习我的NN。

谢谢!

0 个答案:

没有答案