Changeset 1206 for trunk/test/ensemble_test.cc

Timestamp:

Mar 5, 2008, 6:56:01 PM (15 years ago)

Author:

Peter

Message:

fixes #345

File:

• trunk/test/ensemble_test.cc (modified) (3 diffs)

trunk/test/ensemble_test.cc

@@ -32 +32 @@
 #include "yat/classifier/Kernel_MEV.h"
 #include "yat/classifier/MatrixLookup.h"
+#include "yat/classifier/MatrixLookupWeighted.h"
 #include "yat/classifier/NCC.h"
 #include "yat/classifier/PolynomialKernelFunction.h"
@@ -87 +88 @@
     *error << "build ensemble" << std::endl;
     ensemble.build();
+    std::vector<std::vector<statistics::Averager> > result;
+    ensemble.predict(data, result);
+  }
+
+  {
+    *error << "create ensemble of ncc" << std::endl;
+    classifier::MatrixLookupWeighted data_weighted(data);
+    classifier::NCC<statistics::EuclideanDistance> ncc;
+    classifier::CrossValidationSampler sampler(target,3,3);
+    classifier::SubsetGenerator<classifier::MatrixLookupWeighted> 
+      subdata(sampler,data_weighted);
+    classifier::EnsembleBuilder<classifier::SupervisedClassifier,
+      classifier::MatrixLookupWeighted> ensemble(ncc, data_weighted, sampler);
+    *error << "build ensemble" << std::endl;
+    ensemble.build();
+    std::vector<std::vector<statistics::Averager> > result;
+    ensemble.predict(data_weighted, result);
   }
 
@@ -103 +121 @@
   *error << "build ensemble" << std::endl;
   ensemble.build();
+  std::vector<std::vector<statistics::Averager> > result;
+  ensemble.predict(kernel_lookup, result);
   
   utility::Vector out(target.size(),0);