Pytorch datasets.UDPOS.splits 抛出错误

Question

我想按字段将 UDPOS 数据集拆分为训练、有效和测试。下面是我的代码-

import torch
import torch.nn as nn
import torch.optim as optim

from torchtext.legacy import data
from torchtext import datasets

SEED = 1234

random.seed(SEED)
np.random.seed(SEED)
torch.manual_seed(SEED)
torch.backends.cudnn.deterministic = True

TEXT = data.Field(lower = True)
UD_TAGS = data.Field(unk_token = None)
PTB_TAGS = data.Field(unk_token = None)

fields = (("text", TEXT), ("udtags", UD_TAGS), ("ptbtags", PTB_TAGS))

train_data, valid_data, test_data = datasets.UDPOS.splits(fields)

这段代码给我以下错误-

我使用的是 Pytorch 版本 -“1.10.2”。

如何在当前版本中使用 fileds 拆分 UDPOS 数据集。

Answer 1

我通过更改代码解决了同样的问题

from torchtext import datasets

至

from torchtext.legacy import datasets

Pytorch datasets.UDPOS.splits 抛出错误

Pytorch datasets.UDPOS.splits throwing error

nlp

deep-learning

pytorch