Commit f96785af authored by Jordi Inglada's avatar Jordi Inglada

ENH: add function to translate labels to range [0 ... nbClasses-1] for Shark

parent 73056db2
......@@ -30,6 +30,7 @@ otbSEMClassifierNew.cxx
otbDecisionTreeNew.cxx
otbKMeansImageClassificationFilterNew.cxx
otbMachineLearningModelTemplates.cxx
otbSharkUtilsTests.cxx
)
add_executable(otbLearningBaseTestDriver ${OTBLearningBaseTests})
......@@ -68,3 +69,5 @@ otb_add_test(NAME leTuDecisionTreeNew COMMAND otbLearningBaseTestDriver
otb_add_test(NAME leTuKMeansImageClassificationFilterNew COMMAND otbLearningBaseTestDriver
otbKMeansImageClassificationFilterNew)
otb_add_test(NAME leTuSharkNormalizeLabels COMMAND otbLearningBaseTestDriver
otbSharkNormalizeLabels)
......@@ -29,4 +29,5 @@ void RegisterTests()
REGISTER_TEST(otbSEMClassifierNew);
REGISTER_TEST(otbDecisionTreeNew);
REGISTER_TEST(otbKMeansImageClassificationFilterNew);
REGISTER_TEST(otbSharkNormalizeLabels);
}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment