Target Encoder for regression and classification targets.
Each category is encoded based on a shrunk estimate of the average target values for observations belonging to the category. The encoding scheme mixes the global target mean with the target mean conditioned on the value of the category (see [MIC]).
When the target type is “multiclass”, encodings are based on the conditional probability estimate for each class. The target is first binarized using the “one-vs-all” scheme via LabelBinarizer, then the average target value for each class and each category is used for encoding, resulting in n_features * n_classes encoded output features.
TargetEncoder considers missing values, such as np.nan or None, as another category and encodes them like any other category. Categories that are not seen during fit are encoded with the target mean, i.e. target_mean_.
For a demo on the importance of the TargetEncoder internal cross-fitting, see Target Encoder’s Internal Cross fitting. For a comparison of different encoders, refer to Comparing Target Encoder with Other Encoders. Read more in the User Guide.
Note
fit(X, y).transform(X) does not equal fit_transform(X, y) because a cross fitting scheme is used in fit_transform for encoding. See the User Guide for details.
Added in version 1.3.
Categories (unique values) per feature:
"auto" : Determine categories automatically from the training data.
list : categories[i] holds the categories expected in the i-th column. The passed categories should not mix strings and numeric values within a single feature, and should be sorted in case of numeric values.
The used categories are stored in the categories_ fitted attribute.
Type of target.
"auto" : Type of target is inferred with type_of_target.
"continuous" : Continuous target
"binary" : Binary target
"multiclass" : Multiclass target
Note
The type of target inferred with "auto" may not be the desired target type used for modeling. For example, if the target consisted of integers between 0 and 100, then type_of_target will infer the target as "multiclass". In this case, setting target_type="continuous" will specify the target as a regression problem. The target_type_ attribute gives the target type used by the encoder.
Changed in version 1.4: Added the option ‘multiclass’.
The amount of mixing of the target mean conditioned on the value of the category with the global target mean. A larger smooth value will put more weight on the global target mean. If "auto", then smooth is set to an empirical Bayes estimate.
Determines the number of folds in the cross fitting strategy used in fit_transform. For classification targets, StratifiedKFold is used and for continuous targets, KFold is used.
Whether to shuffle the data in fit_transform before splitting into folds. Note that the samples within each split will not be shuffled.
When shuffle is True, random_state affects the ordering of the indices, which controls the randomness of each fold. Otherwise, this parameter has no effect. Pass an int for reproducible output across multiple function calls. See Glossary.
Encodings learnt on all of X. For feature i, encodings_[i] are the encodings matching the categories listed in categories_[i]. When target_type_ is “multiclass”, the encoding for feature i and class j is stored in encodings_[j + (i * len(classes_))]. E.g., for 2 features (f) and 3 classes (c), encodings are ordered: f0_c0, f0_c1, f0_c2, f1_c0, f1_c1, f1_c2,
The categories of each input feature determined during fitting or specified in categories (in order of the features in X and corresponding with the output of transform).
Type of target.
The overall mean of the target. This value is only used in transform to encode categories.
Number of features seen during fit.
n_features_in_,)
Names of features seen during fit. Defined only when X has feature names that are all strings.
If target_type_ is ‘binary’ or ‘multiclass’, holds the label for each class, otherwise None.
See also
OrdinalEncoder
Performs an ordinal (integer) encoding of the categorical features. Contrary to TargetEncoder, this encoding is not supervised. Treating the resulting encoding as a numerical features therefore lead arbitrarily ordered values and therefore typically lead to lower predictive performance when used as preprocessing for a classifier or regressor.
OneHotEncoder
Performs a one-hot encoding of categorical features. This unsupervised encoding is better suited for low cardinality categorical variables as it generate one new feature per unique category.
References
Examples
With smooth="auto", the smoothing parameter is set to an empirical Bayes estimate:
>>> import numpy as np >>> from sklearn.preprocessing import TargetEncoder >>> X = np.array([["dog"] * 20 + ["cat"] * 30 + ["snake"] * 38], dtype=object).T >>> y = [90.3] * 5 + [80.1] * 15 + [20.4] * 5 + [20.1] * 25 + [21.2] * 8 + [49] * 30 >>> enc_auto = TargetEncoder(smooth="auto") >>> X_trans = enc_auto.fit_transform(X, y)
>>> # A high `smooth` parameter puts more weight on global mean on the categorical >>> # encodings: >>> enc_high_smooth = TargetEncoder(smooth=5000.0).fit(X, y) >>> enc_high_smooth.target_mean_ np.float64(44.3) >>> enc_high_smooth.encodings_ [array([44.1, 44.4, 44.3])]
>>> # On the other hand, a low `smooth` parameter puts more weight on target >>> # conditioned on the value of the categorical: >>> enc_low_smooth = TargetEncoder(smooth=1.0).fit(X, y) >>> enc_low_smooth.encodings_ [array([21, 80.8, 43.2])]
Fit the TargetEncoder to X and y.
The data to determine the categories of each feature.
The target data used to encode the categories.
Fitted encoder.
Fit TargetEncoder and transform X with the target encoding.
Note
fit(X, y).transform(X) does not equal fit_transform(X, y) because a cross fitting scheme is used in fit_transform for encoding. See the User Guide. for details.
The data to determine the categories of each feature.
The target data used to encode the categories.
Transformed input.
Get output feature names for transformation.
Not used, present here for API consistency by convention.
Transformed feature names. feature_names_in_ is used unless it is not defined, in which case the following input feature names are generated: ["x0", "x1", ..., "x(n_features_in_ - 1)"]. When type_of_target_ is “multiclass” the names are of the format ‘<feature_name>_<class_name>’.
Get metadata routing of this object.
Please check User Guide on how the routing mechanism works.
A MetadataRequest encapsulating routing information.
Get parameters for this estimator.
If True, will return the parameters for this estimator and contained subobjects that are estimators.
Parameter names mapped to their values.
Set output container.
See Introducing the set_output API for an example on how to use the API.
Configure output of transform and fit_transform.
"default": Default output format of a transformer
"pandas": DataFrame output
"polars": Polars output
None: Transform configuration is unchanged
Added in version 1.4: "polars" option was added.
Estimator instance.
Set the parameters of this estimator.
The method works on simple estimators as well as on nested objects (such as Pipeline). The latter have parameters of the form <component>__<parameter> so that it’s possible to update each component of a nested object.
Estimator parameters.
Estimator instance.
Transform X with the target encoding.
Note
fit(X, y).transform(X) does not equal fit_transform(X, y) because a cross fitting scheme is used in fit_transform for encoding. See the User Guide. for details.
The data to determine the categories of each feature.
Transformed input.
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4