diff --git a/supervised/preprocessing/preprocessing_missing.py b/supervised/preprocessing/preprocessing_missing.py index c6802f78..551d7b4a 100644 --- a/supervised/preprocessing/preprocessing_missing.py +++ b/supervised/preprocessing/preprocessing_missing.py @@ -44,9 +44,12 @@ def _get_fill_value(self, x): PreprocessingMissingValues.MISSING_VALUE ) # add new categorical value return PreprocessingUtils.get_most_frequent(x) - + # datetime if PreprocessingUtils.get_type(x) == PreprocessingUtils.DATETIME: return PreprocessingUtils.get_most_frequent(x) + # text + if PreprocessingUtils.get_type(x) == PreprocessingUtils.TEXT: + return PreprocessingMissingValues.MISSING_VALUE # numerical type if self._na_fill_method == PreprocessingMissingValues.FILL_NA_MIN: