使用 huggingface fill-mask pipeline 获得超过 5 条建议答案

【问题标题】：Using huggingface fill-mask pipeline to get more than 5 suggestions使用 huggingface fill-mask pipeline 获得超过 5 条建议
【发布时间】：2020-07-22 06:24:27
【问题描述】：

下面让我得到 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")

【问题讨论】：

标签： python neural-network nlp huggingface-transformers

【解决方案1】：

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

如果您仔细查看parameters of the FillMaskPipeline（这是pipeline('fill-mask') 的构造，请参阅here），然后你会发现它有一个topk=5 参数，你可以通过在pipeline 构造函数中指定它来简单地将其设置为你喜欢的值：

from transformers import pipeline

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

【讨论】：

作为旁注，我想补充一点，有十个建议通常会导致对进一步建议的置信度得分非常低，但我相信你知道 ;-)

【解决方案2】：

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

【讨论】：