neural-network - 由 PyTorch 中的数字 (csv) 组成的自己的数据集

Question

我想在 PyTorch 中使用我自己的由数字组成的数据集。例如，它们可作为 csv 文件使用。将其加载到 PyTorch 中的最简单方法是什么？到目前为止，我只知道如何在 PyTorch 中使用已经存在的数据集，但我不想这样做。

标签： neural-networkpytorchdataset

Answer 1

导入数据集的最简单方法是：

1. 使用pandas包加载您的 csv 文件：

    import pandas as pd 
    data = pd.read_csv("filename.csv") 

2. 然后，实现一个非常简单的 pytorchDataset类，如此处所述。
3. 您最终会将您的实例Dataset作为 pytorch DataLoader的第一个参数传递