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

问题描述

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

任何帮助将不胜感激!

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)