apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix bug in Naive Bayes test.

Browse Source
This commit is contained in:
Ryan Curtin
2023-11-29 11:11:25 -05:00
parent c1b7694391
commit 2acea2db66
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/tests/nbc_test.cpp
+1 -1
View File
@@ -457,7 +457,7 @@ TEMPLATE_TEST_CASE("NBCIncrementalTest", "[NBCTest]", arma::fmat, arma::mat)
labels[i] = trainData(trainData.n_rows - 1, i);
trainData.shed_row(trainData.n_rows - 1);
NaiveBayesClassifier<MatType> nbc(trainData.n_rows, classes);
NaiveBayesClassifier<MatType> nbc(trainData.n_rows, 2);
for (size_t i = 0; i < trainData.n_cols; ++i)
{