TreeEnsembleClassifier - version 1#

This page documents version 1 of operator TreeEnsembleClassifier. See TreeEnsembleClassifier for the latest version (since version 5).

  • Domain: ai.onnx.ml

  • Since version: 1

Tree Ensemble classifier. Returns the top class for each of N inputs. The attributes named ‘nodes_X’ form a sequence of tuples, associated by index into the sequences, which must all be of equal length. These tuples define the nodes. Similarly, all fields prefixed with `class_` are tuples of votes at the leaves. A leaf may have multiple votes, where each vote is weighted by the associated class_weights index. One and only one of classlabels_strings or classlabels_int64s will be defined. The class_ids are indices into this list.

Inputs

  • X (T1): Input of shape [N,F]

Outputs

  • Y (T2): N, Top class for each point

  • Z (tensor(float)): The class score for each class, for each point, a tensor of shape [N,E].

Type Constraints

  • T1: The input type must be a tensor of a numeric type. Allowed types: tensor(double), tensor(float), tensor(int32), tensor(int64).

  • T2: The output type will be a tensor of strings or integers, depending on which of the classlabels* attributes is used. Allowed types: tensor(int64), tensor(string).