Renaming the base class such as in sklearn

Author: Guillaume Lemaitre

unbalanced_dataset/base_sampler.py

@@ -6,10 +6,10 @@
 
 from ..over_sampling import SMOTE
 from ..under_sampling import EditedNearestNeighbours
-from ..base_sampler import BaseSampler
+from ..base import SamplerMixin
 
 
-class SMOTEENN(BaseSampler):
+class SMOTEENN(SamplerMixin):
     """Class to perform over-sampling using SMOTE and cleaning using ENN.
 
     Combine over- and under-sampling using SMOTE and Edited Nearest Neighbours.

unbalanced_dataset/combine/smote_enn.py

@@ -6,10 +6,10 @@
 
 from ..over_sampling import SMOTE
 from ..under_sampling import TomekLinks
-from ..base_sampler import BaseSampler
+from ..base import SamplerMixin
 
 
-class SMOTETomek(BaseSampler):
+class SMOTETomek(SamplerMixin):
     """Class to perform over-sampling using SMOTE and cleaning using
     Tomek links.
 

unbalanced_dataset/combine/smote_tomek.py

@@ -4,19 +4,19 @@
 
 from abc import ABCMeta, abstractmethod
 
-from ..base_sampler import BaseSampler
+from sklearn.externals import six
 
+from ..base import SamplerMixin
 
-class EnsembleSampler(BaseSampler):
+
+class EnsembleSampler(six.with_metaclass(ABCMeta, SamplerMixin)):
     """Base class for ensenble sampling.
 
     Warning: This class should not be used directly. Use the derive classes
     instead.
 
     """
 
-    __metaclass__ = ABCMeta
-
     @abstractmethod
     def __init__(self, ratio='auto', return_indices=False, random_state=None,
                  verbose=True):

unbalanced_dataset/ensemble/ensemble_sampler.py

@@ -6,10 +6,10 @@
 
 from sklearn.externals import six
 
-from ..base_sampler import BaseSampler
+from ..base import SamplerMixin
 
 
-class OverSampler(six.with_metaclass(ABCMeta, BaseSampler)):
+class OverSampler(six.with_metaclass(ABCMeta, SamplerMixin)):
     """Base class for over-sampling.
 
     Warning: This class should not be used directly. Use the derive classes

unbalanced_dataset/over_sampling/over_sampler.py

@@ -6,10 +6,10 @@
 
 from sklearn.externals import six
 
-from ..base_sampler import BaseSampler
+from ..base import SamplerMixin
 
 
-class UnderSampler(six.with_metaclass(ABCMeta, BaseSampler)):
+class UnderSampler(six.with_metaclass(ABCMeta, SamplerMixin)):
     """Base class for under-sampling.
 
     Warning: This class should not be used directly. Use the derive classes