diff --git a/Parameters.h b/Parameters.h index a9b35f4..1adf252 100644 --- a/Parameters.h +++ b/Parameters.h @@ -25,7 +25,6 @@ namespace masc int maxDepth = 25; //To be left as a parameter of the training plugin (default 25) int minSampleCount = 1; //To be left as a parameter of the training plugin (default 1) int maxCategories = 0; //Normally not important as there’s no categorical variable - bool calcVarImportance = true; //Must be true int activeVarCount = 0; //Use 0 as the default parameter (works best) int maxTreeCount = 100; //Left as a parameter of the training plugin (default: 100) }; diff --git a/q3DMASCClassifier.cpp b/q3DMASCClassifier.cpp index bd3b6a0..15dd8ef 100644 --- a/q3DMASCClassifier.cpp +++ b/q3DMASCClassifier.cpp @@ -462,7 +462,7 @@ bool Classifier::train( const ccPointCloud* cloud, m_rtrees = cv::ml::RTrees::create(); m_rtrees->setMaxDepth(params.maxDepth); m_rtrees->setMinSampleCount(params.minSampleCount); - m_rtrees->setCalculateVarImportance(params.calcVarImportance); + m_rtrees->setCalculateVarImportance(true); m_rtrees->setActiveVarCount(params.activeVarCount); cv::TermCriteria terminationCriteria(cv::TermCriteria::MAX_ITER, params.maxTreeCount, std::numeric_limits::epsilon()); m_rtrees->setTermCriteria(terminationCriteria);