Dont import inf

nielstron · Feb 4, 2019 · cf7cf15 · cf7cf15
1 parent 2c10e07
commit cf7cf15
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/quantulum3/classifier.py b/quantulum3/classifier.py
@@ -7,8 +7,6 @@
 import json
 import logging
 import pkg_resources
-from math import inf
-import numpy
 
 # Semi-dependencies
 try:
@@ -136,7 +134,7 @@ def train_classifier(parameters=None,
             'loss': 'log',
             'penalty': 'l2',
             'max_iter': 50,
-            'tol': -inf,
+            'tol': float("-inf"),
             'alpha': 0.00001,
             'fit_intercept': True
         }