Skip to content

Commit 06e46db

Browse files
Merge pull request #432 from Asiaticum/fix-polynomial-wrapper
Fixed the DataFrame conversion of y in PolynomialWrapper's fit method
2 parents 3d588f6 + 2a1aa8a commit 06e46db

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

category_encoders/wrapper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def __init__(self, feature_encoder: utils.BaseEncoder):
7878
def fit(self, X, y, **kwargs):
7979
# unite the input into pandas types
8080
X, y = utils.convert_inputs(X, y)
81-
y = pd.DataFrame(y, columns=['target'])
81+
y = pd.DataFrame(y.rename('target'))
8282

8383
# apply one-hot-encoder on the label
8484
self.label_encoder = encoders.OneHotEncoder(handle_missing='error',

0 commit comments

Comments
 (0)