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

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

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

任何帮助将不胜感激！