diff --git a/sklearn/preprocessing/_target_encoder.py b/sklearn/preprocessing/_target_encoder.py index 1ae5102afcd59..078a308d7fe6e 100644 --- a/sklearn/preprocessing/_target_encoder.py +++ b/sklearn/preprocessing/_target_encoder.py @@ -339,3 +339,6 @@ def _transform_X_ordinal( for f_idx, encoding in enumerate(encodings): X_out[indices, f_idx] = encoding[X_ordinal[indices, f_idx]] X_out[X_unknown_mask[:, f_idx], f_idx] = y_mean + + def _more_tags(self): + return {"requires_y": True}