如何使用tensorflow saved_model.load

问题描述 投票:0回答:1

我正在按照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"]
tensorflow tensorflow2.0 tensorflow-serving
1个回答
© www.soinside.com 2019 - 2024. All rights reserved.