✨ Add alternative with imblearn.fixes._mode

Author: hayesall

imblearn/over_sampling/_smote/base.py

@@ -19,7 +19,6 @@
 from sklearn.utils import check_array
 from sklearn.utils.sparsefuncs_fast import csr_mean_variance_axis0
 from sklearn.utils.sparsefuncs_fast import csc_mean_variance_axis0
-from sklearn.utils.fixes import _mode
 
 from ..base import BaseOverSampler
 from ...metrics.pairwise import ValueDifferenceMetric
@@ -29,6 +28,7 @@
 from ...utils._docstring import _n_jobs_docstring
 from ...utils._docstring import _random_state_docstring
 from ...utils._validation import _deprecate_positional_args
+from ...utils.fixes import _mode
 
 
 class BaseSMOTE(BaseOverSampler):

imblearn/under_sampling/_prototype_selection/_edited_nearest_neighbours.py

@@ -11,13 +11,14 @@
 import numpy as np
 
 from sklearn.utils import _safe_indexing
-from sklearn.utils.fixes import _mode
 
 from ..base import BaseCleaningSampler
 from ...utils import check_neighbors_object
 from ...utils import Substitution
 from ...utils._docstring import _n_jobs_docstring
 from ...utils._validation import _deprecate_positional_args
+from ...utils.fixes import _mode
+
 
 SEL_KIND = ("all", "mode")
 

imblearn/under_sampling/_prototype_selection/_neighbourhood_cleaning_rule.py

@@ -9,14 +9,15 @@
 import numpy as np
 
 from sklearn.utils import _safe_indexing
-from sklearn.utils.fixes import _mode
 
 from ..base import BaseCleaningSampler
 from ._edited_nearest_neighbours import EditedNearestNeighbours
 from ...utils import check_neighbors_object
 from ...utils import Substitution
 from ...utils._docstring import _n_jobs_docstring
 from ...utils._validation import _deprecate_positional_args
+from ...utils.fixes import _mode
+
 
 SEL_KIND = ("all", "mode")
 

imblearn/utils/fixes.py

@@ -0,0 +1,26 @@
+"""Compatibility fixes for older version of python, numpy and scipy
+If you add content to this file, please give the version of the package
+at which the fix is no longer needed.
+
+Backdated from scikit-learn.
+"""
+# Authors: Emmanuelle Gouillart <emmanuelle.gouillart@normalesup.org>
+#          Gael Varoquaux <gael.varoquaux@normalesup.org>
+#          Fabian Pedregosa <fpedregosa@acm.org>
+#          Lars Buitinck
+#
+# License: BSD 3 clause
+
+from sklearn.externals._packaging.version import parse as parse_version
+import scipy
+import scipy.stats
+
+
+sp_version = parse_version(scipy.__version__)
+
+
+# TODO: Remove when SciPy 1.9 is the minimum supported version
+def _mode(a, axis=0):
+    if sp_version >= parse_version("1.9.0"):
+        return scipy.stats.mode(a, axis=axis, keepdims=True)
+    return scipy.stats.mode(a, axis=axis)