How to export/save a text classifier after fine-tuned on a pre-trained BERT

Question

I followed the steps on the Predicting Movie Reviews with BERT on TF Hub here.

At the end, how do I export the model to be used/loaded as a classifier later?

I found a link that shows that I can export the estimator as a tf.saved_model. However, I got stuck on creating the 'serving_input_receiver_fn()'.

@AlbertoBonsanto Thanks for your input. Could you elaborate on that? Do you mean dump the 'estimator' using pickle? — ebaik, Mar 29 '19 at 15:55
Check this out. It's a helpful tips https://stackoverflow.com/a/63347334/9690325 — Emeka Boris Ama, Sep 22 '20 at 10:52

score 0 · Answer 1 · answered Sep 22 '20 at 10:56

0

Use pickle to save the model

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'))

answered Sep 22 '20 at 10:56

Emeka Boris Ama

429
4
5

How to export/save a text classifier after fine-tuned on a pre-trained BERT

1 Answers1