如何使用训练有素的 bert 模型预测句子

时间:2021-03-14 19:31:48

标签: python tensorflow text-classification bert-language-model

我是使用 BERT 和 TensorFlow 的初学者,这可能是一个愚蠢的问题。我已经使用 BERT 教程完成了句子分类,该教程可以在这个 google colab notebook here 中找到。本教程唯一遗漏的部分是 model.predict 部分,您可以在其中输入随机句子以查看模型如何对它们进行分类(可能使用 argmax 来获得最接近的值)。到目前为止,我还没有弄清楚如何在我自己随机输入的句子上应用分词器,也没有想出如何对使用的数据集具有的两种类型的类应用预测。

任何帮助将不胜感激!

0 个答案:

没有答案