Change double to float

scikit-learn-contrib · Nov 21, 2023 · 6138f39 · 6138f39
1 parent 0be576d
commit 6138f39
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/category_encoders/cat_boost.py b/category_encoders/cat_boost.py
@@ -103,7 +103,7 @@ def __init__(self, verbose=0, cols=None, drop_invariant=False, return_df=True,
 
     def _fit(self, X, y, **kwargs):
         X = X.copy(deep=True)
-        y = y.astype('double') #Incase y is bool.
+        y = y.astype(float) #Incase y is bool or categorical.
         self._mean = y.mean()
         self.mapping = {col: self._fit_column_map(X[col], y) for col in self.cols}