diff --git a/Modules/Applications/AppClassification/app/otbTrainLibSVM.cxx b/Modules/Applications/AppClassification/app/otbTrainLibSVM.cxx index 458e5b8487b8f6e11c337c3ffc0026941c8ca627..e117337708d3effd267a462a1d24f8b2f6736e18 100644 --- a/Modules/Applications/AppClassification/app/otbTrainLibSVM.cxx +++ b/Modules/Applications/AppClassification/app/otbTrainLibSVM.cxx @@ -42,6 +42,9 @@ namespace Wrapper AddParameter(ParameterType_Empty, "classifier.libsvm.opt", "Parameters optimization"); MandatoryOff("classifier.libsvm.opt"); SetParameterDescription("classifier.libsvm.opt", "SVM parameters optimization flag."); + AddParameter(ParameterType_Empty, "classifier.libsvm.prob", "Probability estimation"); + MandatoryOff("classifier.libsvm.prob"); + SetParameterDescription("classifier.libsvm.prob", "Probability estimation flag."); } @@ -56,6 +59,10 @@ namespace Wrapper { libSVMClassifier->SetParameterOptimization(true); } + if (IsParameterEnabled("classifier.libsvm.prob")) + { + libSVMClassifier->SetDoProbabilityEstimates(true); + } libSVMClassifier->SetC(GetParameterFloat("classifier.libsvm.c")); switch (GetParameterInt("classifier.libsvm.k"))