torch.utils.data.Dataset用法

利用繼承自Dataset的類，可以訪問訓練所需的數據
比如一下數據：保存爲csv文件

from torch.utils.data import Dataset
import pandas as pd #這個包用來讀取CSV數據
class mydataset(Dataset):
    def __init__(self,csv_file): #self參數必須，其他參數及其形式隨程序需要而不同，比如(self,*inputs)
        self.csv_data=pd.read_csv(csv_file)
    def __len__(self):
        return len(self.csv_data)
    def __getitem__(self,idx):
        data=self.csv_data.values[idx]
        return data
data=mydataset('/home/yls/Documents/test.csv')
print(data[3])
print(len(data))

輸出結果如下：
[‘OpenSuse’ ‘stable’ ‘OpenSuse Repository’
‘zypper\xa0in\xa0python3-pandas’]
6