Skip to content

Commit

Permalink
fix #1461
Browse files Browse the repository at this point in the history
  • Loading branch information
guolinke authored Jun 20, 2018
1 parent 277f7e4 commit c184852
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/objective/objective_function.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,8 @@ ObjectiveFunction* ObjectiveFunction::CreateObjectiveFunction(const std::string&
return new CrossEntropy(strs);
} else if (type == std::string("xentlambda") || type == std::string("cross_entropy_lambda")) {
return new CrossEntropyLambda(strs);
} else if (type == std::string("mean_absolute_percentage_error") || type == std::string("mape")) {
return new RegressionMAPELOSS(strs);
} else if (type == std::string("gamma")) {
return new RegressionGammaLoss(strs);
} else if (type == std::string("tweedie")) {
Expand Down

0 comments on commit c184852

Please sign in to comment.