diff options
Diffstat (limited to 'tests/common/ClassifierTests.cc')
-rw-r--r-- | tests/common/ClassifierTests.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/common/ClassifierTests.cc b/tests/common/ClassifierTests.cc index d950304..693f744 100644 --- a/tests/common/ClassifierTests.cc +++ b/tests/common/ClassifierTests.cc @@ -35,7 +35,7 @@ void test_classifier_result(std::vector<std::pair<uint32_t, T>>& selectedResults } arm::app::Classifier classifier; - REQUIRE(classifier.GetClassificationResults(outputTensor, resultVec, labels, 5)); + REQUIRE(classifier.GetClassificationResults(outputTensor, resultVec, labels, 5, true)); REQUIRE(5 == resultVec.size()); for (size_t i = 0; i < resultVec.size(); ++i) { @@ -50,7 +50,7 @@ TEST_CASE("Common classifier") TfLiteTensor* outputTens = nullptr; std::vector <arm::app::ClassificationResult> resultVec; arm::app::Classifier classifier; - REQUIRE(!classifier.GetClassificationResults(outputTens, resultVec, {}, 5)); + REQUIRE(!classifier.GetClassificationResults(outputTens, resultVec, {}, 5, true)); } SECTION("Test classification results") |