Skip to content

GitLab

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

informative error for custom obj in RF (#2355)

parent 0dfda826
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/boosting/rf.hpp
View file @ 76e57c5d
...@@ -81,7 +81,7 @@ class RF : public GBDT { ...@@ -81,7 +81,7 @@ class RF : public GBDT {
void Boosting() override { void Boosting() override {
if (objective_function_ == nullptr) { if (objective_function_ == nullptr) {
Log::Fatal("No object function provided"); Log::Fatal("RF mode do not support custom objective function, please use built-in objectives.");
} }
init_scores_.resize(num_tree_per_iteration_, 0.0); init_scores_.resize(num_tree_per_iteration_, 0.0);
for (int cur_tree_id = 0; cur_tree_id < num_tree_per_iteration_; ++cur_tree_id) { for (int cur_tree_id = 0; cur_tree_id < num_tree_per_iteration_; ++cur_tree_id) {
......
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