使用huggingface fillmask管道获得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

同样，这是“在建”文档的一个不幸缺陷

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

from transformers import pipeline

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