TreeFeaturizer Class
TreeFeaturizer.
- Inheritance
-
nimbusml.internal.core.feature_extraction._treefeaturizer.TreeFeaturizerTreeFeaturizernimbusml.base_transform.BaseTransformTreeFeaturizersklearn.base.TransformerMixinTreeFeaturizer
Constructor
TreeFeaturizer(predictor_model, suffix=None, label_permutation_seed=0, columns=None, **params)Parameters
| Name | Description |
|---|---|
|
columns
|
see Columns. |
|
predictor_model
|
Trainer to use. |
|
suffix
|
Output column: The suffix to append to the default column names. |
|
label_permutation_seed
|
If specified, determines the permutation seed for applying this featurizer to a multiclass problem. |
|
params
|
Additional arguments sent to compute engine. |
Remarks
Trains a tree ensemble, or loads it from a file, then maps a numeric feature vector to three outputs:
A vector containing the individual tree outputs of the tree ensemble.
A vector indicating the leaves that the feature vector falls on in the tree ensemble.
A vector indicating the paths that the feature vector falls on in the tree ensemble. If a both a model file and a trainer are specified, will use the model file. If neither are specified, will train a default FastTree model. This can handle key labels by training a regression model towards their optionally permuted indices.
Methods
| get_params |
Get the parameters for this operator. |
get_params
Get the parameters for this operator.
get_params(deep=False)Parameters
| Name | Description |
|---|---|
|
deep
|
Default value: False
|