Changeset 1864

Timestamp:

Mar 15, 2009, 3:49:53 AM (15 years ago)

Author:

Peter

Message:

adding test for train and predict. refs #269

File:

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

trunk/test/svm_multi_class_test.cc

@@ -22 +22 @@
 #include "Suite.h"
 
+#include "yat/classifier/Kernel_SEV.h"
+#include "yat/classifier/KernelLookup.h"
+#include "yat/classifier/MatrixLookupWeighted.h"
+#include "yat/classifier/PolynomialKernelFunction.h"
 #include "yat/classifier/SvmMultiClass.h"
+
+#include <fstream>
+#include <string>
 
 using namespace theplu::yat;
 
 void test_construction(test::Suite&);
+void test_predict(test::Suite&);
 
 int main( int argc, char* argv[])
@@ -33 +41 @@
   suite.err() << "testing SvmMultiClass" << std::endl;
   test_construction(suite);
+  test_predict(suite);
 
   return suite.return_value();
@@ -53 +62 @@
   delete svm3;
 }
+
+
+void test_predict(test::Suite& suite)
+{
+  using namespace classifier;
+  std::string file = test::filename("data/sorlie_centroid_data.txt");
+  std::ifstream is(file.c_str());
+  suite.err() << "load data `" << file << "'" << std::endl;
+  MatrixLookupWeighted data(is, '\t');
+  is.close();
+  PolynomialKernelFunction linear;
+  suite.err() << "calculating kernel" << std::endl;
+  Kernel_SEV kernel_raw(data, linear);
+  KernelLookup kernel(kernel_raw);
+  file = test::filename("data/sorlie_centroid_classes.txt");
+  suite.err() << "load classes `" << file << "'" << std::endl;
+  is.open(file.c_str());
+  Target target(is);
+  is.close();
+
+  SvmMultiClass svm;
+  suite.err() << "training svm" << std::endl;
+  svm.train(kernel, target);
+
+  utility::Matrix result;
+  svm.predict(kernel, result);
+
+  if (!suite.add(result.rows()==5 && result.columns()==79)) {
+    suite.err() << "ERROR: incorrect dimension in result Matrix\n"
+                << "found " << result.rows() << "x" << result.columns() << "\n"
+                << "expected 5x79\n";
+  }
+
+  // we expect perfect predictions on training data
+  for (size_t i=0; i<79; ++i) {
+    for (size_t j=0; j<5; ++j) {
+      if (target(i)==j && result(j,i)<0) {
+        suite.err() << "result(" << j << "," << i << ") is "
+                    << result(j,i) << " expected greater than 0" << std::endl;
+        suite.add(false);
+      }
+      else if (target(i)!=j && result(j,i)>0) {
+        suite.err() << "result(" << j << "," << i << ") is "
+                    << result(j,i) << " expected smaller than 0" << std::endl;
+        suite.add(false);
+      }
+    }
+  }
+}