From 591d428f3fb90ba4e44b8ffd70e9b8355a84ceea Mon Sep 17 00:00:00 2001 From: Ryan Curtin Date: Thu, 16 Nov 2017 15:08:26 -0500 Subject: [PATCH] Build the random forest Python binding, provide better warning output. --- src/mlpack/methods/random_forest/CMakeLists.txt | 1 + src/mlpack/methods/random_forest/random_forest_main.cpp | 3 +++ 2 files changed, 4 insertions(+) diff --git a/src/mlpack/methods/random_forest/CMakeLists.txt b/src/mlpack/methods/random_forest/CMakeLists.txt index 1521dddc69..3ac70de8a8 100644 --- a/src/mlpack/methods/random_forest/CMakeLists.txt +++ b/src/mlpack/methods/random_forest/CMakeLists.txt @@ -18,3 +18,4 @@ endforeach() set(MLPACK_SRCS ${MLPACK_SRCS} ${DIR_SRCS} PARENT_SCOPE) add_cli_executable(random_forest) +add_python_binding(random_forest) diff --git a/src/mlpack/methods/random_forest/random_forest_main.cpp b/src/mlpack/methods/random_forest/random_forest_main.cpp index 176d450095..c32c7bb3f7 100644 --- a/src/mlpack/methods/random_forest/random_forest_main.cpp +++ b/src/mlpack/methods/random_forest/random_forest_main.cpp @@ -101,6 +101,9 @@ void mlpackMain() RequireParamValue("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")) {