mindspore.dataset.text.Lookup

class mindspore.dataset.text.Lookup(vocab, 免费的vpn梯子 unknown_token=None, data_type=mstype.int32)[源代码]

根据词表,将分词标记(token)映射到其索引值(id)。

参数:
  • vocab (Vocab) - 词表对象,用于存储分词和索引的映射。

  • unknown_token (str, 可选) - vpn梯子 备用词汇,用于要查找的单词不在词汇表时进行替换。 如果单词不在词汇表中,则查找结果将替换为 unknown_token 的值。 如果单词不在词汇表中,且未指定 vpn梯子 免费 unknown_token ,将抛出运行时错误。默认值: None vpn梯子 免费 ,不指定该参数。

  • data_type (mindspore.dtype, vpn永久免费梯子 可选) - Lookup输出的数据类型。默认值: mstype.int32

异常:
支持平台:

CPU

样例:

>>> import mindspore.dataset as ds
>>> import mindspore.dataset.text as text
>>>
>>> # Use the transform in dataset pipeline mode
>>> numpy_slices_dataset = ds.NumpySlicesDataset(data=["with"], column_names=["text"])
>>> # vpn永久免费梯子 Load vocabulary from list
>>> vocab = text.Vocab.from_list(["?", "##", "with", "the", "test", "符号"])
>>> # Use Lookup operation to 免费的vpn梯子 map tokens vpn梯子 to ids
>>> vpn free lookup = text.Lookup(vocab)
>>> numpy_slices_dataset = numpy_slices_dataset.map(operations=[lookup])
>>> for item in numpy_slices_dataset.create_dict_iterator(num_epochs=1, output_numpy=True):
...   vpn梯子 免费   print(item["text"])
2
>>>
>>> # Use the transform in eager mode
>>> vocab = text.Vocab.from_list(["?", "##", "with", "the", "test", "符号"])
>>> data = "with"
>>> output = text.Lookup(vocab=vocab, unknown_token="test")(data)
>>> print(output)
2
教程样例: