使用bert的NLP意外测试

Question

我一直在使用bert建模进行惊奇。我需要做的是 使用这些句子是这样的：

The dog sitting next to the chairs [MASK]

并尝试测试“面具”是否“是”和“被”的意外状态。

from transformers import pipeline
unmasker=pipeline('fill-mask',model='bert-base-uncased')
unmasker("The dog sitting next to the chairs [MASK].")

但是我想要的是缩小“是”和“是”之间的选择。