Changeset 156

Timestamp:

Sep 10, 2004, 3:34:43 PM (19 years ago)

Author:

Peter

Message:

.

File:

• trunk/test/test_roc.cc (modified) (2 diffs)

trunk/test/test_roc.cc

@@ -61 +61 @@
   
   theplu::gslapi::vector correct_area(3);
-  correct_area(0)=1.0;
-  correct_area(1)=0.75;
-  correct_area(2)=0.50;
+  correct_area(0)=8.0/9.0;
+  correct_area(1)=6.0/9.0;
+  correct_area(2)=1.0;
 
+  const double tol = 0.001;
   for (size_t i=0; i<data.rows(); i++){
     theplu::gslapi::vector vec = data[i];
     area = roc.score(target2,vec);
-    if (area!=correct_area(i)){
+    if (area<correct_area(i)-tol || area>correct_area(i)+tol){
       cerr << "test_roc: area is " << area << " should be " 
            << correct_area(i) << endl;
@@ -75 +76 @@
   }
 
+  theplu::gslapi::vector weight(target2.size(),1);
+  for (size_t i=0; i<data.rows(); i++){
+    theplu::gslapi::vector vec = data[i];
+    area = roc.score(target2, vec, weight);
+    if (area<correct_area(i)-tol || area>correct_area(i)+tol){
+      cerr << "test_roc: weighted area is " << area << " should be " 
+           << correct_area(i) << endl;
+      return -1;
+    }
+  }
+
+
   if (ok)
     return 0;