CategoricalImputer: Error out when no mode is found

Author: dukebody

sklearn_pandas/categorical_imputer.py

@@ -64,7 +64,12 @@ def fit(self, X, y=None):
         mask = _get_mask(X, self.missing_values)
         X = X[~mask]
 
-        self.fill_ = Counter(X).most_common(1)[0][0]
+        modes = pd.Series(X).mode()
+        if modes.shape[0] == 0:
+            raise ValueError('No value is repeteated more than '
+                             'twice in the column')
+        else:
+            self.fill_ = modes[0]
 
         return self
 

tests/test_categorical_imputer.py

@@ -34,6 +34,20 @@ def test_unit(input_type, none_value):
     assert (Xt == ['a', 'b', 'b', 'b']).all()
 
 
+@pytest.mark.parametrize('input_type', ['np', 'pd'])
+def test_no_mode(input_type):
+
+    data = ['a', 'b', 'c', np.nan]
+
+    if input_type == 'pd':
+        X = pd.Series(data)
+    else:
+        X = np.asarray(data, dtype=object)
+
+    with pytest.raises(ValueError):
+        CategoricalImputer().fit_transform(X)
+
+
 @pytest.mark.parametrize('input_type', ['np', 'pd'])
 def test_missing_values_param(input_type):