在经过预训练的BERT上进行微调后，如何导出/保存文本分类器

Question

我在以下TF Hub上按照BERT预测电影评论的步骤进行了操作：https : //colab.research.google.com/github/google-research/bert/blob/master/predicting_movie_reviews_with_bert_on_tf_hub.ipynb#scrollTo=PPVEXhNjYXC-。

最后，如何导出模型以供以后用作分类器？

我找到了一个链接（https://guillaumegenthial.github.io/serving-tensorflow-estimator.html），该链接表明我可以将估算器导出为tf.saved_model。但是，我在创建“ serving_input_receiver_fn（）”时遇到了麻烦。

Answer 1

使用pickle保存模型

import pickle
# save the model to disk
filename = 'finalized_model.sav'
pickle.dump(model, open(filename, 'wb'))

#Load the model
loaded_model = pickle.load(open(filename, 'rb'))