Skip to content

Commit

Permalink
Merge 7ad3f8d into 0b2a93d
Browse files Browse the repository at this point in the history
  • Loading branch information
StrikerRUS committed May 6, 2020
2 parents 0b2a93d + 7ad3f8d commit c8ee8ce
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
3 changes: 2 additions & 1 deletion m2cgen/assemblers/boosting.py
Expand Up @@ -66,7 +66,8 @@ def _assemble_multi_class_output(self, estimator_params):
return ast.VectorVal(proba_exprs)

def _assemble_bin_class_output(self, estimator_params):
# Base score is calculated based on https://github.com/dmlc/xgboost/blob/master/src/objective/regression_loss.h#L64 # noqa
# Base score is calculated based on
# https://github.com/dmlc/xgboost/blob/8de7f1928e4815843fbf8773a5ac7ecbc37b2e15/src/objective/regression_loss.h#L91
# return -logf(1.0f / base_score - 1.0f);
base_score = 0
if self._base_score != 0:
Expand Down
2 changes: 1 addition & 1 deletion m2cgen/cli.py
Expand Up @@ -111,7 +111,7 @@ def generate_code(args):
# the same as the default value of the keyword argument of the exporter
# (this is due to languages like C# which prefer their method names to
# follow PascalCase unlike all the other supported languages -- see
# https://github.com/BayesWitnesses/m2cgen/pull/166/files#r379867601
# https://github.com/BayesWitnesses/m2cgen/pull/166#discussion_r379867601
# for more).
if arg_name == 'function_name' and arg_value is None:
param = inspect.signature(exporter).parameters['function_name']
Expand Down

0 comments on commit c8ee8ce

Please sign in to comment.