Skip to content

GitLab

Unverified Commit c184852b authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

fix #1461

parent 277f7e47
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/objective/objective_function.cpp
View file @ c184852b
...@@ -73,6 +73,8 @@ ObjectiveFunction* ObjectiveFunction::CreateObjectiveFunction(const std::string& ...@@ -73,6 +73,8 @@ ObjectiveFunction* ObjectiveFunction::CreateObjectiveFunction(const std::string&
return new CrossEntropy(strs); return new CrossEntropy(strs);
} else if (type == std::string("xentlambda") || type == std::string("cross_entropy_lambda")) { } else if (type == std::string("xentlambda") || type == std::string("cross_entropy_lambda")) {
return new CrossEntropyLambda(strs); 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")) { } else if (type == std::string("gamma")) {
return new RegressionGammaLoss(strs); return new RegressionGammaLoss(strs);
} else if (type == std::string("tweedie")) { } else if (type == std::string("tweedie")) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment