Использование huggingface fill-mask pipe для получения более 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

Опять же, это прискорбный недостаток документации, находящейся в стадии разработки.

Если вы внимательно посмотрите на параметры FillMaskPipeline (что и составляет конструкция pipeline('fill-mask'), см. здесь ), тогда вы обнаружите, что он имеет параметр topk=5, который можно просто установить на значение по своему вкусу, указав его в конструкторе pipeline:

from transformers import pipeline

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