Convert encoding to dict

ClaudioSalvatoreArcidiacono · Jan 2, 2025 · 7e6be52 · 7e6be52
1 parent cc84a2d
commit 7e6be52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sklearo/encoding/base.py b/sklearo/encoding/base.py
@@ -109,7 +109,7 @@ def fit(self, X: IntoFrameT, y: IntoSeriesT) -> "BaseTargetEncoder":
                     )
                 if self.unseen == "fill" and self.fill_value_unseen == "mean":
                     self.mean_target_[class_] = X_y_binarized[target_col_name].mean()
-
+            self.encoding_map_ = dict(self.encoding_map_)
         else:
             for column in self.columns_:
                 self.encoding_map_[column] = self._calculate_target_statistic(