首页 > 解决方案 > 打印 torch.data.dataset 对象的值

问题描述

我已经将我的 pandas 数据框X_train变成了z张量,但输出是

user_id=torchtext.data.RawField()
fields=[('user_id',user_id)]
from torchtext.data import Dataset,Example
z=torchtext.data.Dataset(X_train.user_id,fields)
print(len(z))
print(z)

输出是:

426018
<torchtext.data.dataset.Dataset object at 0x7feffb6a8f98>

我应该如何在我的变量对象 z 中打印实际数据?

标签: pythonpytorch

解决方案


可能print(list(torch.utils.data.DataLoader()))是你要找的吗?

DataLoader(dataset, batch_size=1, shuffle=False, sampler=None,
           batch_sampler=None, num_workers=0, collate_fn=None,
           pin_memory=False, drop_last=False, timeout=0,
           worker_init_fn=None)

推荐阅读