Merge 508c709 into 024b61e

BayesWitnesses · May 27, 2020 · a1370db · a1370db
2 parents 024b61e + 508c709
commit a1370db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/m2cgen/assemblers/boosting.py b/m2cgen/assemblers/boosting.py
@@ -1,4 +1,5 @@
 import json
+import math
 import numpy as np
 
 from m2cgen import ast
@@ -73,7 +74,7 @@ def _assemble_bin_class_output(self, estimator_params):
         # return -logf(1.0f / base_score - 1.0f);
         base_score = 0
         if self._base_score != 0:
-            base_score = -np.log(1.0 / self._base_score - 1.0)
+            base_score = -math.log(1.0 / self._base_score - 1.0)
 
         expr = self._assemble_single_output(
             estimator_params, base_score=base_score)