From e11658cab1fbcb92b72833f1bd3619064979b8f0 Mon Sep 17 00:00:00 2001 From: Guillaume Pasero Date: Thu, 30 Jul 2015 16:56:26 +0200 Subject: [PATCH] ENH: expose probability option on libsvm model --- .../Applications/AppClassification/app/otbTrainLibSVM.cxx | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/Modules/Applications/AppClassification/app/otbTrainLibSVM.cxx b/Modules/Applications/AppClassification/app/otbTrainLibSVM.cxx index 458e5b8487..e117337708 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")) -- GitLab