使用ONNX Runtime运行Keras转换的ONNX多标签文本分类模型时遇feed格式问题
问题:ONNX多标签文本分类模型推理的Feed格式确认
我基于Keras官方多标签文本分类示例训练并转换得到了.onnx格式的多标签文本分类模型,该模型接收文本输入并输出预测类别。目前参考keras-onnx的MNIST教程进行推理,但不确定如何确定推理时feed的格式。我的Keras模型结构如下:
def make_model(): shallow_mlp_model = keras.Sequential( [ layers.Dense(512, activation="relu"), layers.Dense(256, activation="relu"), layers.Dense(lookup.vocabulary_size(), activation="sigmoid"), ] ) return shallow_mlp_model
内容的提问来源于stack exchange,提问作者lm231
相关产品推荐
相关产品推荐

