Skip to content

Commit 661fb5d

Browse files
author
florian
committed
Set parameters in BinaryEncoder.__init__
as instance variables. -> Necessary for it to work in an sklearn.ColumnTransformer!
1 parent 01c833a commit 661fb5d

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

category_encoders/binary.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,9 +71,16 @@ class BinaryEncoder(BaseEstimator, TransformerMixin):
7171

7272
def __init__(self, verbose=0, cols=None, mapping=None, drop_invariant=False, return_df=True,
7373
handle_unknown='value', handle_missing='value'):
74-
self.base_n_encoder = ce.BaseNEncoder(base=2, verbose=verbose, cols=cols, mapping=mapping,
75-
drop_invariant=drop_invariant, return_df=return_df,
76-
handle_unknown=handle_unknown, handle_missing=handle_missing)
74+
self. verbose = verbose
75+
self.cols = cols
76+
self.mapping = mapping
77+
self.drop_invariant = drop_invariant
78+
self.return_df = return_df
79+
self.handle_unknown = handle_unknown
80+
self.handle_missing = handle_missing
81+
self.base_n_encoder = ce.BaseNEncoder(base=2, verbose=self.verbose, cols=self.cols, mapping=self.mapping,
82+
drop_invariant=self.drop_invariant, return_df=self.return_df,
83+
handle_unknown=self.handle_unknown, handle_missing=self.handle_missing)
7784

7885
def fit(self, X, y=None, **kwargs):
7986
"""Fit encoder according to X and y.

0 commit comments

Comments
 (0)