Changeset 335 for trunk/test/nni_test.cc

Timestamp:

Jun 2, 2005, 11:46:55 PM (18 years ago)

Author:

Jari Häkkinen

Message:

Added WeNNI test with binary weights (i.e. kNNI mode).

File:

• trunk/test/nni_test.cc (modified) (1 diff)

trunk/test/nni_test.cc

@@ -102 +102 @@
           ok=false; // calucaltion error detected
         }
-  if(ok)
-    return 0;
-  return -1;
+  control_stream.close();
+  data_stream.close();
+  weight_stream.close();
+
+  // test WeNNI with binary weights
+  data_stream.open(knni_data.c_str());
+  data=gslapi::matrix(data_stream);
+  weight_stream.open(knni_weight.c_str());
+  weight=gslapi::matrix(weight_stream);
+  utility::WeNNI wenni2(data,weight,neighbours);
+  wenni2.estimate();
+  control_stream.open(knni_result.c_str());
+  control=gslapi::matrix(control_stream);
+  control-=wenni2.imputed_data();
+  for (unsigned int i=0; i<control.rows(); i++)
+    for (unsigned int j=0; j<control.columns(); j++)
+      // Jari, should we use GSL defined round off errors? Anyway, the
+      // hardcoded number below should be changed.
+      if (fabs(control(i,j))>error_bound) {
+        if (print)
+          std::cerr << "WeNNI binary weight test FAILED.\nError on row " << i
+                    << " and column " << j << " is " << fabs(control(i,j))
+                    << ". Expected less than " << error_bound << std::endl;
+        ok=false; // calculation result out of accepted error bounds
+      }
+      else
+        if (!((control(i,j)==control(i,j)))) {
+          if (print) 
+            std::cerr << "WeNNI binary wieght test FAILED.\n"
+                      << "nan (not a number) encountered in test on row " << i
+                      << " and column " << j << std::endl;
+          ok=false; // calucaltion error detected
+        }
+  control_stream.close();
+  data_stream.close();
+  weight_stream.close();
+
+  return (ok ? 0 : -1);
 }