Skip to content

Commit f1e4899

Browse files
UPD: label encoder standalone method
1 parent dd4dd9d commit f1e4899

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

mapie/classification.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -909,6 +909,16 @@ def _get_classes_info(
909909

910910
return n_classes, classes
911911

912+
def _get_label_encoder(self) -> LabelEncoder:
913+
"""
914+
Construct the label encoder with respect to the classes values.
915+
916+
Returns
917+
-------
918+
LabelEncoder
919+
"""
920+
return LabelEncoder().fit(self.classes_)
921+
912922
def _check_fit_parameter(
913923
self,
914924
X: ArrayLike,
@@ -972,11 +982,9 @@ def _check_fit_parameter(
972982
n_samples = _num_samples(y)
973983

974984
self.n_classes_, self.classes_ = self._get_classes_info(estimator, y)
975-
enc = LabelEncoder()
976-
enc.fit(self.classes_)
977-
y_enc = enc.transform(y)
985+
self.label_encoder_ = self._get_label_encoder()
986+
y_enc = self.label_encoder_.transform(y)
978987

979-
self.label_encoder_ = enc
980988
self._check_target(y)
981989

982990
return estimator, cv, X, y, y_enc, sample_weight, groups, n_samples

0 commit comments

Comments
 (0)