Skip to content

GitLab

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

Allow feature_fraction=1.0 in random forest mode

parent cc99f0d3
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/boosting/rf.hpp
View file @ adc1004f
...@@ -27,7 +27,7 @@ public: ...@@ -27,7 +27,7 @@ public:
void Init(const Config* config, const Dataset* train_data, const ObjectiveFunction* objective_function, void Init(const Config* config, const Dataset* train_data, const ObjectiveFunction* objective_function,
const std::vector<const Metric*>& training_metrics) override { 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->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); GBDT::Init(config, train_data, objective_function, training_metrics);
if (num_init_iteration_ > 0) { if (num_init_iteration_ > 0) {
......
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