mlpack
Classes | Macros | Functions
random_forest_test.cpp File Reference
#include <mlpack/core.hpp>
#include <mlpack/core/util/mlpack_main.hpp>
#include <mlpack/methods/random_forest/random_forest_main.cpp>
#include "test_helper.hpp"
#include "../catch.hpp"
#include "../test_catch_tools.hpp"
Include dependency graph for random_forest_test.cpp:

Classes

struct  RandomForestTestFixture
 

Macros

#define BINDING_TYPE   BINDING_TYPE_TEST
 

Functions

 TEST_CASE_METHOD (RandomForestTestFixture, "RandomForestOutputDimensionTest", "[RandomForestMainTest][BindingTests]")
 Check that number of output points and number of input points are equal and have appropriate number of classes.
 
 TEST_CASE_METHOD (RandomForestTestFixture, "RandomForestModelReuseTest", "[RandomForestMainTest][BindingTests]")
 Ensure that saved model can be used again.
 
 TEST_CASE_METHOD (RandomForestTestFixture, "RandomForestNumOfTreesTest", "[RandomForestMainTest][BindingTests]")
 Make sure number of trees specified is always a positive number.
 
 TEST_CASE_METHOD (RandomForestTestFixture, "RandomForestMinimumLeafSizeTest", "[RandomForestMainTest][BindingTests]")
 Make sure minimum leaf size specified is always a positive number.
 
 TEST_CASE_METHOD (RandomForestTestFixture, "RandomForestMaximumDepthTest", "[RandomForestMainTest][BindingTests]")
 Make sure maximum depth specified is always a positive number.
 
 TEST_CASE_METHOD (RandomForestTestFixture, "RandomForestTrainingVerTest", "[RandomForestMainTest][BindingTests]")
 Make sure only one of training data or pre-trained model is passed, when warm_start is not passed.
 
template<typename TreeType >
bool CheckDifferentTrees (const TreeType &nodeA, const TreeType &nodeB)
 
 TEST_CASE_METHOD (RandomForestTestFixture, "RandomForestDiffMinLeafSizeTest", "[RandomForestMainTest][BindingTests]")
 Ensure that the trees have different structure as the minimum leaf size is changed.
 
 TEST_CASE_METHOD (RandomForestTestFixture, "RandomForestDiffNumTreeTest", "[RandomForestMainTest][BindingTests]")
 Ensure that the number of trees are different when num_trees is specified differently.
 
 TEST_CASE_METHOD (RandomForestTestFixture, "RandomForestDiffMaxDepthTest", "[RandomForestMainTest][BindingTests]")
 Ensure that the maximum_depth parameter makes a difference.
 
 TEST_CASE_METHOD (RandomForestTestFixture, "RandomForestTrainingModelWarmStart", "[RandomForestMainTest][BindingTests]")
 Make sure that training and input_model are both passed when warm_start is false.
 
 TEST_CASE_METHOD (RandomForestTestFixture, "RandomForestWarmStart", "[RandomForestMainTest][BindingTests]")
 Ensuring that model does gets trained on top of existing one when warm_start and input_model are both passed.
 

Detailed Description

Author
Manish Kumar

Test mlpackMain() of random_forest_main.cpp.

mlpack is free software; you may redistribute it and/or modify it under the terms of the 3-clause BSD license. You should have received a copy of the 3-clause BSD license along with mlpack. If not, see http://www.opensource.org/licenses/BSD-3-Clause for more information.