mlpack
|
#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"
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... | |
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.
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 | ( | 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.