Address the problem of ratio

glemaitre · Jun 7, 2016 · 01ba4c1 · 01ba4c1
1 parent 843bc85
commit 01ba4c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/unbalanced_dataset/base_sampler.py b/unbalanced_dataset/base_sampler.py
@@ -50,9 +50,9 @@ def __init__(self, ratio='auto', random_state=None, verbose=True):
         # over the number of samples in the majority class. Thus, the ratio
         # cannot be greater than 1.0
         if isinstance(ratio, float):
-            if ratio > 1.0:
+            if ratio > 1:
                 raise ValueError('Ration cannot be greater than one.')
-            elif ratio < 0.:
+            elif ratio <= 0:
                 raise ValueError('Ratio cannot be negative.')
             else:
                 self.ratio_ = ratio