Allow feature_fraction=1.0 in random forest mode

microsoft · Nov 30, 2018 · adc1004 · adc1004
1 parent cc99f0d
commit adc1004
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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<const Metric*>& 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) {