From ef52e858ae6eaf9d290adfa747735ccb80a68678 Mon Sep 17 00:00:00 2001 From: Rishabh Garg <56191449+RishabhGarg108@users.noreply.github.com> Date: Sat, 27 Mar 2021 21:24:05 +0530 Subject: [PATCH] Added suggestion --- src/mlpack/methods/random_forest/random_forest_main.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/mlpack/methods/random_forest/random_forest_main.cpp b/src/mlpack/methods/random_forest/random_forest_main.cpp index f64fdc8c07..c73385a709 100644 --- a/src/mlpack/methods/random_forest/random_forest_main.cpp +++ b/src/mlpack/methods/random_forest/random_forest_main.cpp @@ -207,9 +207,9 @@ static void mlpackMain() ReportIgnoredParam({{ "training", false }}, "minimum_leaf_size"); RandomForestModel* rfModel; - // Handles the case when we are either training on top of existing forest - // else we are making predictions only. - if (IO::HasParam("warm_start") or IO::HasParam("input_model")) + // Input model is loaded when we are either doing warm-started training or + // else we are making predictions only or both. + if (IO::HasParam("input_model")) rfModel = IO::GetParam("input_model"); // Handles the case when we are training new forest from scratch. else