3

我可以使用拥抱脸的蒙版填充管道来预测句子中的 1 个蒙面标记，使用以下代码：

!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")

但是，如果我想预测 2 个蒙面标记，是否有人对执行此操作的最佳方法有意见？例如，如果句子是相反的"I am going to <mask> <mask> in this sentence"？

如果我尝试将这个确切的句子放入 nlp_fill 我会收到错误"ValueError: only one element tensors can be converted to Python scalars"，因此它不会自动工作。

任何帮助将非常感激！

4

1 回答 1

2

可悲的是，掩码后面只有一个单词的期望被硬编码到 FillMaskPipeline 类中。

对于一个更受限制的问题，给定固定数量的掩码填充选项来填充掩码，可以扩展FitBERT来执行此操作 - 如果您给我发消息，我可以发送给您一个笔记本，但这是非常糟糕的代码。我是 FitBERT 的作者之一，但我没有机会添加这个。

编辑

这是笔记本，不要评判我

于 2020-05-21T21:19:40.997 回答