基础使用
首先简单介绍Dataset 和 DataLoader 两个类,Dataset直接对接了数据源,主要实现两个函数 getlen() 和 getitems().
Dataset 作为DataLoader的参数,Dataloader 可以配置batch_size 等等取样参数,用于获得每批次的数据。
from torch.utils.data import *
text_dataset = TensorDataset(X, y)
loader= DataLoader(dataset=text_dataset