diff --git a/Modules/Applications/AppClassification/app/otbVectorRegression.cxx b/Modules/Applications/AppClassification/app/otbVectorRegression.cxx
index f6490ff4f28fa4fd98a26fbdbf515f26c178a30f..74ac136552b34fd6aa65f2996f0bfd8532f95a29 100644
--- a/Modules/Applications/AppClassification/app/otbVectorRegression.cxx
+++ b/Modules/Applications/AppClassification/app/otbVectorRegression.cxx
@@ -90,6 +90,7 @@ VectorRegression
   SetOfficialDocLink();
 }
 
+// Confidence map computation is not support for regression.
 template<>
 bool
 VectorRegression
diff --git a/Modules/Applications/AppClassification/include/otbVectorPrediction.hxx b/Modules/Applications/AppClassification/include/otbVectorPrediction.hxx
index 1fe65059bfd638ab24dba04b05cfd1c2837b2a04..8178735bcb2b728393a563a537576d73f3963332 100644
--- a/Modules/Applications/AppClassification/include/otbVectorPrediction.hxx
+++ b/Modules/Applications/AppClassification/include/otbVectorPrediction.hxx
@@ -34,7 +34,14 @@ VectorPrediction <RegressionMode>
 ::DoInit()
 {
   DoInitSpecialization();
-  //TODO add assert to check that parameters has been correctly defined
+  
+  // Assert that the all needed parameters have ben definied in DoInitSpecialization
+  assert(GetParameterByKey("in") != nullptr);
+  assert(GetParameterByKey("instat") != nullptr);
+  assert(GetParameterByKey("model") != nullptr);
+  assert(GetParameterByKey("cfield") != nullptr);
+  assert(GetParameterByKey("feat") != nullptr);
+  assert(GetParameterByKey("out") != nullptr);
 }
 
 template <bool RegressionMode>
@@ -316,7 +323,6 @@ VectorPrediction <RegressionMode>
 
   clock_t toc = clock();
   otbAppLogINFO( "Elapsed: "<< ((double)(toc - tic) / CLOCKS_PER_SEC)<<" seconds.");
-
 }
 
 } //end namespace wrapper