如何通过Hugging Face Fill-Mask Pipeline获取10个掩码词建议？

当然可以啦！Hugging Face的Fill-Mask Pipeline默认确实只返回5个候选掩码标记，但你只需要简单设置一个参数就能拿到10个结果，非常容易实现。

你可以通过两种方式来调整返回的候选数量：

• 初始化Pipeline时指定参数：在创建pipeline实例的时候，直接传入top_k=10，这样后续所有调用都会默认返回10个候选：

!pip install -q transformers
from __future__ import print_function
import ipywidgets as widgets
from transformers import pipeline

# 初始化时设置top_k为10
nlp_fill = pipeline('fill-mask', top_k=10)
# 调用时会返回10个掩码建议
results = nlp_fill("I am going to guess <mask> in this sentence")
print(results)

• 调用时临时传递参数：如果只是某一次调用需要返回10个结果，也可以在调用方法的时候单独传入top_k=10，不影响其他调用的默认行为：

!pip install -q transformers
from __future__ import print_function
import ipywidgets as widgets
from transformers import pipeline

nlp_fill = pipeline('fill-mask')
# 本次调用返回10个掩码建议
results = nlp_fill("I am going to guess <mask> in this sentence", top_k=10)
print(results)

top_k参数就是用来控制返回的候选掩码标记数量的，你可以根据需求设置成任意正整数，只要模型能够生成足够多的合理候选即可。

内容的提问来源于stack exchange，提问作者user3472360