ai.onnx.ml - LinearClassifier

LinearClassifier - 1 (ai.onnx.ml)

版本

  • name: LinearClassifier (GitHub)

  • 域: ai.onnx.ml

  • 起始版本: 1

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

该算子的此版本**自 ai.onnx.ml 域的 1 版本起**可用。

摘要

线性分类器

属性

  • classlabels_ints - INTS :

    使用整数标签时的类别标签。必须定义“classlabels”属性，且只能定义一个。

  • classlabels_strings - STRINGS :

    使用字符串标签时的类别标签。必须定义“classlabels”属性，且只能定义一个。

  • coefficients - FLOATS (必需)

    模型权重集合。

  • intercepts - FLOATS :

    截距集合。

  • multi_class - INT (默认值为 '0')

    指示是执行 OvR（一对多）还是多项式分类（0=OvR 为默认值）。

  • post_transform - STRING (默认值为 'NONE')

    指示要应用于得分向量的变换。
    可以是 ‘NONE,’ ‘SOFTMAX,’ ‘LOGISTIC,’ ‘SOFTMAX_ZERO,’ 或 ‘PROBIT’ 之一

输入

  • X (异构) - T1

    要分类的数据。

输出

  • Y (异构) - T2

    分类输出（每个样本一个类别）。

  • Z (异构) - tensor(float)

    分类得分（[N,E] - 每个类别和每个样本一个得分

类型约束

  • T1 in ( tensor(double), tensor(float), tensor(int32), tensor(int64) )

    输入必须是数值类型的张量，形状为 [N,C] 或 [C]。后一种情况将被视为 [1,C]

  • T2 在 ( tensor(int64), tensor(string) ) 中

    输出将是字符串或整数类型的张量。