使用bert tensorflow估算器进行保存，加载和预测

Question

我尝试按照此链接上的答案保存模型。

Saving and doing Inference with Tensorflow BERT model

尽管我能够保存并重新加载模型。我无法使用已加载的模型进行预测。我用于加载模型的代码如下。

from tensorflow.contrib import predictor

predict_fn = predictor.from_saved_model("./savedModel/1580195293")   #the exported model dir

我不确定用于预测的格式，即predict_fn应该接收参数的格式，或者我使用的代码不正确。 < / p>

有人可以帮忙吗？

IMDB电影评论示例，类似于我的用例。（仅供参考） https://colab.research.google.com/github/google-research/bert/blob/master/predicting_movie_reviews_with_bert_on_tf_hub.ipynb#scrollTo=ERkTE8-7oQLZ