diff --git a/m2cgen/assemblers/boosting.py b/m2cgen/assemblers/boosting.py index 04ee12e3..27ee3f33 100644 --- 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)