使用 huggingface fill-mask pipeline 获得超过 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

同样，这是 "under construction" 文档的一个不幸的缺点。

如果你仔细观察 parameters of the FillMaskPipeline (which is what pipeline('fill-mask') constructs, see here)，然后你会发现它有一个 topk=5 参数，你可以通过在 pipeline 构造函数中指定它来简单地将它设置为你喜欢的值：

from transformers import pipeline

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

Answer 2

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

使用 huggingface fill-mask pipeline 获得超过 5 个建议

Using huggingface fill-mask pipeline to get more than 5 suggestions

python

nlp

neural-network

huggingface-transformers