java调用文本分类textrnn模型，勿踩坑

最新推荐文章于 2024-05-14 02:34:55 发布

Jeu

最新推荐文章于 2024-05-14 02:34:55 发布

阅读量893

点赞数

分类专栏： java 人工智能文本分类文章标签： java调用深度学习模型

本文链接：https://blog.youkuaiyun.com/cuipanguo/article/details/98881299

版权

直接贴代码
# 将模型保存为可用于线上服务的文件（一个.pb文件，一个variables文件夹）
# print('Exporting trained model to', save_dir)
builder = tf.saved_model.builder.SavedModelBuilder(save_dir)

# 服务器专用代码

classification_signature = (
    tf.saved_model.signature_def_utils.build_signature_def(
        inputs={
            # "image"
            "input_x":
                tf.saved_model.utils.build_tensor_info(rnn.input_x),
            "dropout_keep_prob":
                tf.saved_model.utils.build_tensor_info(rnn.dropout_keep_prob)
        },
        outputs={
            # "classify"
            "output":
                tf.saved_model.utils.build_tensor_info(rnn.predictions)
            # "classification_outputs_scores":
            #     tf.saved_model.utils.build_tensor_info(model.logits)
        },
        # Prediction method name used in a SignatureDef.
        # PREDICT_METHOD_NAME = "tensorflow/serving/predict