使用huggingface fill-mask管道获得超过5条建议

Question

下面让我得到 5 个关于蒙面令牌的建议，但我想得到 10 个建议 - 有谁知道这是否可以通过拥抱脸来实现？

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

nlp_fill = pipeline('fill-mask')
nlp_fill("I am going to guess <mask> in this sentence")

Answer 1

我想补充一下，参数已更改为top_k. 它可以传递给每个单独的调用nlp_fill以及pipeline方法。

Answer 2

这又是“正在建设中”文档的一个不幸的缺点。

如果仔细观察 的参数FillMaskPipeline（构造函数pipeline('fill-mask')，请参见此处），那么您会发现它有一个topk=5参数，您可以通过在构造函数中指定它来将其设置为您喜欢的值pipeline：

from transformers import pipeline

nlp_fill = pipeline('fill-mask', topk=10)