利用繼承自Dataset的類，可以訪問訓(xùn)練所需的數(shù)據(jù)
比如一下數(shù)據(jù)：保存為csv文件

from torch.utils.data import Dataset import pandas as pd #這個(gè)包用來讀取CSV數(shù)據(jù) class mydataset(Dataset):def __init__(self,csv_file): #self參數(shù)必須，其他參數(shù)及其形式隨程序需要而不同，比如(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))

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

總結(jié)

以上是生活随笔為你收集整理的torch.utils.data.Dataset用法的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。