Build the random forest Python binding, provide better warning output.

This commit is contained in:
Ryan Curtin
2017-11-16 15:08:26 -05:00
parent 3691f18fa3
commit 591d428f3f
2 changed files with 4 additions and 0 deletions
@@ -18,3 +18,4 @@ endforeach()
set(MLPACK_SRCS ${MLPACK_SRCS} ${DIR_SRCS} PARENT_SCOPE)
add_cli_executable(random_forest)
add_python_binding(random_forest)
@@ -101,6 +101,9 @@ void mlpackMain()
RequireParamValue<int>("minimum_leaf_size", [](int x) { return x > 0; }, true,
"minimum leaf size must be greater than 0");
ReportIgnoredParam({{ "training", false }}, "num_trees");
ReportIgnoredParam({{ "training", false }}, "minimum_leaf_size");
RandomForestModel rfModel;
if (CLI::HasParam("training"))
{