diff --git a/src/boosting/rf.hpp b/src/boosting/rf.hpp index de0578754e67..bf9b004741a7 100644 --- a/src/boosting/rf.hpp +++ b/src/boosting/rf.hpp @@ -27,7 +27,7 @@ class RF: public GBDT { void Init(const Config* config, const Dataset* train_data, const ObjectiveFunction* objective_function, const std::vector& training_metrics) override { CHECK(config->bagging_freq > 0 && config->bagging_fraction < 1.0f && config->bagging_fraction > 0.0f); - CHECK(config->feature_fraction < 1.0f && config->feature_fraction > 0.0f); + CHECK(config->feature_fraction <= 1.0f && config->feature_fraction > 0.0f); GBDT::Init(config, train_data, objective_function, training_metrics); if (num_init_iteration_ > 0) {