|
@@ -4,6 +4,7 @@
|
|
#include "core/basics/StringTools.h"
|
|
#include "core/basics/StringTools.h"
|
|
|
|
|
|
#include "vislearning/classifier/vclassifier/VCAmitSVM.h"
|
|
#include "vislearning/classifier/vclassifier/VCAmitSVM.h"
|
|
|
|
+#include "vislearning/classifier/vclassifier/VCNearestClassMean.h"
|
|
#include "vislearning/classifier/vclassifier/VCSimpleGaussian.h"
|
|
#include "vislearning/classifier/vclassifier/VCSimpleGaussian.h"
|
|
#include "vislearning/classifier/vclassifier/VCNearestNeighbour.h"
|
|
#include "vislearning/classifier/vclassifier/VCNearestNeighbour.h"
|
|
#include "vislearning/classifier/vclassifier/VCCrossGeneralization.h"
|
|
#include "vislearning/classifier/vclassifier/VCCrossGeneralization.h"
|
|
@@ -66,6 +67,9 @@ class GenericClassifierSelection
|
|
#ifdef NICE_USELIB_ICE
|
|
#ifdef NICE_USELIB_ICE
|
|
} else if ( classifier_type == "gauss" ) {
|
|
} else if ( classifier_type == "gauss" ) {
|
|
classifier = new VCSimpleGaussian( conf );
|
|
classifier = new VCSimpleGaussian( conf );
|
|
|
|
+
|
|
|
|
+ } else if ( classifier_type == "nearest_classmean2" ) {
|
|
|
|
+ classifier = new VCNearestClassMean( conf, new NICE::EuclidianDistance<double>() );
|
|
#endif
|
|
#endif
|
|
} else if ( classifier_type == "random_forest" ) {
|
|
} else if ( classifier_type == "random_forest" ) {
|
|
FeaturePoolClassifier *fpc = new FPCRandomForests ( conf, "RandomForest" );
|
|
FeaturePoolClassifier *fpc = new FPCRandomForests ( conf, "RandomForest" );
|