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

Classes

struct  NBCTestFixture
 

Macros

#define BINDING_TYPE   BINDING_TYPE_TEST
 

Functions

 TEST_CASE_METHOD (NBCTestFixture, "NBCOutputDimensionTest", "[NBCMainTest][BindingTests]")
 Ensure that we get desired dimensions when both training data and labels are passed.
 
 TEST_CASE_METHOD (NBCTestFixture, "NBCLabelsLessDimensionTest", "[NBCMainTest][BindingTests]")
 Check that last row of input file is used as labels when labels are not passed specifically and results are same from both label and labeless models.
 
 TEST_CASE_METHOD (NBCTestFixture, "NBCModelReuseTest", "[NBCMainTest][BindingTests]")
 Ensure that saved model can be used again.
 
 TEST_CASE_METHOD (NBCTestFixture, "NBCTrainingVerTest", "[NBCMainTest][BindingTests]")
 Make sure only one of training data or pre-trained model is passed.
 
 TEST_CASE_METHOD (NBCTestFixture, "NBCIncrementalVarianceTest", "[NBCMainTest][BindingTests]")
 Check that models trained with or without incremental variance outputs same results.
 
 TEST_CASE_METHOD (NBCTestFixture, "NBCOptionConsistencyTest", "[NBCMainTest][BindingTests]")
 Ensure that the parameter 'output' and the parameter 'predictions' give the same output. More...
 
 TEST_CASE_METHOD (NBCTestFixture, "NBCOptionConsistencyTest2", "[NBCMainTest][BindingTests]")
 This test ensures that the parameter 'output_probabilities' and the parameter 'probabilities' give the same output. More...
 

Detailed Description

Author
Manish Kumar

Test mlpackMain() of nbc_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.

Function Documentation

◆ TEST_CASE_METHOD() [1/2]

TEST_CASE_METHOD ( NBCTestFixture  ,
"NBCOptionConsistencyTest"  ,
""  [NBCMainTest][BindingTests] 
)

Ensure that the parameter 'output' and the parameter 'predictions' give the same output.

This test case should be removed in mlpack 4 when the deprecated parameter 'output' is removed.

◆ TEST_CASE_METHOD() [2/2]

TEST_CASE_METHOD ( NBCTestFixture  ,
"NBCOptionConsistencyTest2"  ,
""  [NBCMainTest][BindingTests] 
)

This test ensures that the parameter 'output_probabilities' and the parameter 'probabilities' give the same output.

This test case should be removed in mlpack 4 when the deprecated parameter: 'output_probabilities' is removed.