ai.onnx.ml - OneHotEncoder

OneHotEncoder - 1 (ai.onnx.ml)

Version

  • name: OneHotEncoder (GitHub)

  • domain: ai.onnx.ml

  • since_version: 1

  • function: False

  • support_level: SupportType.COMMON

  • shape inference: True

This version of the operator has been available since version 1 of domain ai.onnx.ml.

Summary

Replace each input element with an array of ones and zeros, where a single one is placed at the index of the category that was passed in. The total category count will determine the size of the extra dimension of the output array Y.
For example, if we pass a tensor with a single value of 4, and a category count of 8, the output will be a tensor with [0,0,0,0,1,0,0,0].
This operator assumes every input feature is from the same set of categories.
If the input is a tensor of float, int32, or double, the data will be cast to integers and the cats_int64s category list will be used for the lookups.

Attributes

  • cats_int64s - INTS :

    List of categories, ints.
    One and only one of the ‘cats_*’ attributes must be defined.

  • cats_strings - STRINGS :

    List of categories, strings.
    One and only one of the ‘cats_*’ attributes must be defined.

  • zeros - INT (default is '1'):

    If true and category is not present, will return all zeros; if false and a category if not found, the operator will fail.

Inputs

  • X (heterogeneous) - T:

    Data to be encoded.

Outputs

  • Y (heterogeneous) - tensor(float):

    Encoded output data, having one more dimension than X.

Type Constraints

  • T in ( tensor(double), tensor(float), tensor(int32), tensor(int64), tensor(string) ):

    The input must be a tensor of a numeric type.