You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

使用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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.21 01:48:38