From f9551364cfbb97fe60f78eea87bbf7846dd564fe Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9dric=20Traizet?= <traizetc@cesbio.cnes.fr>
Date: Mon, 17 Jul 2017 14:06:00 +0200
Subject: [PATCH] debugging vect app
---
.../include/otbMachineLearningModel.h | 16 ++++++++++++----
1 file changed, 12 insertions(+), 4 deletions(-)
diff --git a/Modules/Learning/LearningBase/include/otbMachineLearningModel.h b/Modules/Learning/LearningBase/include/otbMachineLearningModel.h
index 1de069ab6d..7da0aaad12 100644
--- a/Modules/Learning/LearningBase/include/otbMachineLearningModel.h
+++ b/Modules/Learning/LearningBase/include/otbMachineLearningModel.h
@@ -116,8 +116,10 @@ public:
*/
TargetSampleType Predict(const InputSampleType& input, ConfidenceValueType *quality = ITK_NULLPTR) const;
-
- virtual unsigned int GetDimension() {return 1;}; /// This method is used to determine the output vector size after dimensionality reduction, and should be overrided for all machine learning models used for dimensionality reduction. This method is not used for classification and regression
+ itkSetMacro(Dimension,unsigned int);
+ itkGetMacro(Dimension,unsigned int);
+
+ // virtual unsigned int GetDimension() {return 1;}; /// This method is used to determine the output vector size after dimensionality reduction, and should be overrided for all machine learning models used for dimensionality reduction. This method is not used for classification and regression
/** Predict a batch of samples (InputListSampleType)
@@ -186,7 +188,9 @@ protected:
/** Input list sample */
typename InputListSampleType::Pointer m_InputListSample;
-
+
+ typename InputListSampleType::Pointer m_ValidationListSample;
+
/** Target list sample */
typename TargetListSampleType::Pointer m_TargetListSample;
@@ -195,6 +199,9 @@ protected:
/** flag to choose between classification and regression modes */
bool m_RegressionMode;
+ /** Output Dimension of the model, used by Dimensionality Reduction models*/
+
+
/** flag that indicates if the model supports regression, child
* classes should modify it in their constructor if they support
* regression mode */
@@ -205,7 +212,8 @@ protected:
/** Is DoPredictBatch multi-threaded ? */
bool m_IsDoPredictBatchMultiThreaded;
-
+ unsigned int m_Dimension;
+
private:
/** Actual implementation of BatchPredicition
* Default implementation will call DoPredict iteratively
--
GitLab