Skip to content
Snippets Groups Projects
Commit f9551364 authored by Cédric Traizet's avatar Cédric Traizet
Browse files

debugging vect app

parent 7b717e3a
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment