From e11658cab1fbcb92b72833f1bd3619064979b8f0 Mon Sep 17 00:00:00 2001
From: Guillaume Pasero <guillaume.pasero@c-s.fr>
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