torchtext.data.utils

get_tokenizer(tokenizer, language=‘en’)

功能:使用分词器对句子进行分词。

参数:

  • tokenizer:分词器名称。

    如果传入None,功能相当于simple_space_split()不会分割标点符号

    如果传入"basic_english",则会将字母转为小写并按空格分割,会分割标点符号

    如果传入可调用函数,则会调用并返回相应结果。

    如果传入分词库名称,则会返回相应的结果,分词库有spacy、moses、toktok、revotk,subword等。

  • language:语言名称,默认为en

示例:

import torchtext
from torchtext.data.utils import get_tokenizer

# 传入None
tokenizer = get_tokenizer(None)
tokens = tokenizer("You can,\t\tnow\n\ninstall TorchText using pip!!!")
print(tokens)

# 传入"basic_english"
tokenizer = get_tokenizer("basic_english")
tokens = tokenizer("You can,\t\tnow\n\ninstall TorchText using pip!!!")
print(tokens)

# 传入可调用函数
def mySplit(text:str):
    return text.split(" ")

tokenizer = get_tokenizer(mySplit)
tokens = tokenizer("You can,\t\tnow\n\ninstall TorchText using pip!!!")
print(tokens)

# 传入分词库名称
tokenizer = get_tokenizer("moses")
tokens = tokenizer("You can,\t\tnow\n\ninstall TorchText using pip!!!")
print(tokens)

示例结果:

['You', 'can,', 'now', 'install', 'TorchText', 'using', 'pip!!!']
['you', 'can', ',', 'now', 'install', 'torchtext', 'using', 'pip', '!', '!', '!']
['You', 'can,\t\tnow\n\ninstall', 'TorchText', 'using', 'pip!!!']
['You', 'can', ',', 'now', 'install', 'TorchText', 'using', 'pip', '!', '!', '!']

ngarms_iterator(token_list, ngrams)

功能:生成ngram词袋分词。

扫描二维码关注公众号,回复: 16189698 查看本文章

参数:

  • token_list:分词结果列表。
  • ngrams:词袋大小

示例:

import torchtext
from torchtext.data.utils import get_tokenizer, ngrams_iterator

tokenizer = get_tokenizer("basic_english")
tokens = tokenizer("You can\t\tnow")

list(ngrams_iterator(tokens, 3))

示例结果:

['you', 'can', 'now', 'you can', 'can now', 'you can now']

猜你喜欢

转载自blog.csdn.net/qq_42464569/article/details/120791006