فهرست منبع

Added method to return model parameters of linear regression.

Frank Prüfer 11 سال پیش
والد
کامیت
a5435f267c
2فایلهای تغییر یافته به همراه7 افزوده شده و 6 حذف شده
  1. 4 6
      regression/linregression/LinRegression.cpp
  2. 3 0
      regression/linregression/LinRegression.h

+ 4 - 6
regression/linregression/LinRegression.cpp

@@ -103,15 +103,13 @@ void LinRegression::teach ( const NICE::VVector & x, const NICE::Vector & y ){
     params.multiply(tmpAlphaInv,rhs);
 
     alpha = params.std_vector();
-    
-    cerr<<"Alpha-Vektor: ";
-    for(uint i=0;i<alpha.size();i++){
-      cerr<<alpha[i]<<" ";
-    }
-    cerr<<endl;
   }
 }
 
+std::vector<double> LinRegression::getModelParams(){
+  return alpha;
+}
+
 double LinRegression::predict ( const NICE::Vector & x ){
   double y;
   if ( dim == 2 ){  //two-dimensional least squares

+ 3 - 0
regression/linregression/LinRegression.h

@@ -39,6 +39,9 @@ class LinRegression : public RegressionAlgorithm
     /** method to learn model parameters */
     void teach ( const NICE::VVector & x, const NICE::Vector & y );
     
+    /** returns model parameters as a vector */
+    std::vector<double> getModelParams();
+    
     /** method to predict function value */
     double predict ( const NICE::Vector & x );
 };