相关疑难解决方法(0)

我正在尝试保存一个经过微调的 bert 模型。我已经正确运行了代码 - 它工作正常，并且在 ipython 控制台中我可以调用 getPrediction 并让它产生结果。

我保存了体重文件（最高的是 model.ckpt-333.data-00000-of-00001

我不知道如何保存模型以供重复使用。

我正在使用 bert-tensorflow。

import json

import pandas as pd
import tensorflow as tf
import tensorflow_hub as hub
from datetime import datetime


from sklearn.model_selection import train_test_split
import os

print("tensorflow version : ", tf.__version__)
print("tensorflow_hub version : ", hub.__version__)


#Importing BERT modules
import bert
from bert import run_classifier
from bert import optimization
from bert import tokenization

#set output directory of the model
OUTPUT_DIR = 'model'

#@markdown Whether or not to clear/delete …