Merge pull request #354 from PaulWestenthanner/fix/issue_353

fixed bug in LOO encoder reported in #353

Author: PaulWestenthanner

category_encoders/leave_one_out.py

@@ -243,7 +243,8 @@ def transform_leave_one_out(self, X, y, mapping=None):
             is_unknown_value = X[col].isin(unseen_values.dropna().astype(object))
 
             if X[col].dtype.name == 'category': # Pandas 0.24 tries hard to preserve categorical data type
-                X[col] = X[col].astype(str)
+                index_dtype = X[col].dtype.categories.dtype
+                X[col] = X[col].astype(index_dtype)
 
             if self.handle_unknown == 'error' and is_unknown_value.any():
                 raise ValueError('Columns to be encoded can not contain new values')

docs/requirements.txt

@@ -1,3 +1,4 @@
+sphinx>=3.0
 sphinx_rtd_theme
 numpydoc
 numpy>=1.14.0