时间戳

首页 > 解决方案 > 使用 huggingface fill-mask pipeline 获得超过 5 条建议

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

问题描述

下面让我为蒙面令牌获得 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")

标签： pythonneural-networknlphuggingface-transformers

解决方案

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

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

from transformers import pipeline

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

推荐阅读