Changeset 731 for trunk/test/regression_test.cc

Timestamp:

Jan 6, 2007, 5:06:19 PM (16 years ago)

Author:

Peter

Message:

added test for multidimensional weighted and straight version

File:

• trunk/test/regression_test.cc (modified) (4 diffs)

trunk/test/regression_test.cc

@@ -43 +43 @@
 bool equal(regression::OneDimensional&, regression::OneDimensionalWeighted&, 
            std::ostream*); 
+
+bool multidim(std::ostream* error);
 
 bool unity_weights(regression::OneDimensional&, 
@@ -185 +187 @@
   *error << "  testing regression::PolynomialWeighted" << std::endl;
   regression::PolynomialWeighted pol_weighted(2);
+  // ok = equal(polynomial, pol_weighted, error) && ok;
+
+  ok = multidim(error) && ok;
 
   if (error!=&std::cerr)
@@ -220 +225 @@
   if (r.predict(2000) != rw.predict(2000)){
     ok = false;
-    *error << "Error: predict not equal" << std::endl;
+    *error << "Error: predict not equal\n" 
+           << "   weighted: " << rw.predict(2000) << "\n"
+           << "   non-weighted: " << r.predict(2000)
+           << std::endl;
   }
   if (fabs(r.prediction_error2(2000)-rw.prediction_error2(2000))>10e-7){
@@ -340 +348 @@
 
 
+bool multidim(std::ostream* error)
+{
+  bool ok = true;
+  *error << "  testing regression::MultiDimensionalWeighted" << std::endl;
+  utility::vector x(5); x(0)=1970; x(1)=1980; x(2)=1990; x(3)=2000; x(4)=2010;
+  utility::vector y(5); y(0)=12;   y(1)=11;   y(2)=14;   y(3)=13;   y(4)=15;
+  utility::vector w(5,1.0);
+  
+  utility::matrix data(5,3);
+  for (size_t i=0; i<data.rows(); ++i){
+    data(i,0)=1;
+    data(i,1)=x(i);
+    data(i,2)=x(i)*x(i);
+  }
+  regression::MultiDimensional md;
+  md.fit(data,y);
+  regression::MultiDimensionalWeighted mdw;
+  mdw.fit(data,y,w);
+  utility::vector z(3,1);
+  z(1)=2000;
+  z(2)=2000*2000;
+  if (md.predict(z) != mdw.predict(z)){
+    ok = false;
+    *error << "Error: predict not equal\n" 
+           << "   weighted: " << mdw.predict(1) << "\n"
+           << "   non-weighted: " << md.predict(1)
+           << std::endl;
+  }
+  /*
+  if (md.standard_error2(z) != mdw.standard_error2(z)){
+    ok = false;
+    *error << "Error: standard_error2 not equal\n" 
+           << "   weighted: " << mdw.standard_error2(z) << "\n"
+           << "   non-weighted: " << md.standard_error2(z)
+           << std::endl;
+  }
+  */
+  return ok;
+}
+
+
 bool Local_test(regression::OneDimensionalWeighted& r, 
                 regression::Kernel& k)