使用文本分类 model 进行预测

Question

我基本上遵循了以下代码： https://towardsdatascience.com/fine-tuning-pre-trained-transformer-models-for-sentence-entailment-d87caf9ec9db 。

model 接受了以下数据的训练：

sentence 1: "Clearly, California can't do better."
sentence 2: "California must do better."
label: "Contradiction"

我最终得到了 model：“模型”和从转换器库导入的标记器“BertTokenizer”。

现在我想在一组数据上使用我的 model：

sentence 1: "Example sentence 1"
sentence 2: "Example sentence 2"

我希望我的 model 到 output label 基于它的预测。 那么，我现在如何从一对句子中的 go 给它们一个关系 label 呢？