Commit 9f03bc4b authored by Jorge Samper Gonzalez's avatar Jorge Samper Gonzalez

corrected bug in apply_best_parameter in RandomForest

parent 5b3e158f
Pipeline #769 passed with stages
in 14 seconds
......@@ -533,9 +533,9 @@ class RandomForest(base.MLAlgorithm):
if result_feat is None:
max_features = 1.0
elif result_feat in ["auto", "sqrt"]:
max_features = np.sqrt(n_features)
max_features = np.sqrt(n_features) / n_features
elif result_feat == "log2":
max_features = np.log2(n_features)
max_features = np.log2(n_features) / n_features
elif isinstance(result_feat, int):
max_features = float(result_feat) / n_features
elif isinstance(result_feat, float):
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