我正在按照this official tensorflow tutorial建立文本分类模型
我正在这样输出经过训练的模型
serving_input_fn = tf.estimator.export.build_parsing_serving_input_receiver_fn(tf.feature_column.make_parse_example_spec([embedded_text_feature_column]))
export_path = estimator.export_saved_model("./models/sentiment", serving_input_fn)
我不确定在加载时如何传递示例句子(例如“那是一部很棒的电影”)来进行预测。
imported = tf.saved_model.load(b'./models/sentiment/1586848142')
infer = imported.signatures["serving_default"]