lregress.h

来自「统计软件包」· C头文件 代码 · 共 142 行

#ifndef LREGRESS_H_
#define LREGRESS_H_

#include <vector>

/**
  Linear regression and related functions

  The lregress class packages the lr function, which calculates a
  linear regression line given paired vectors of X and Y points.


<h4>
   Copyright and Use
</h4>

   You may use this source code without limitation and without
   fee as long as you include:

<p>
<i>
     This software was written and is copyrighted by Ian Kaplan, Bear
     Products International, www.bearcave.com, 2001.
</i>
</p>

   This software is provided "as is", without any warranty or
   claim as to its usefulness.  Anyone who uses this source code
   uses it at their own risk.  Nor is any support provided by
   Ian Kaplan and Bear Products International.

   Please send any bug fixes or suggested source changes to:

<pre>
     [e-mail]
</pre>

 */
class lregress
{
private:
  lregress( const lregress &rhs );

public:
  lregress() {}
  ~lregress() {}

  /**
     A container for an {x,y} point of type double.
   */
  class point
  {
  private:
    double x_, y_;
  public:
    /** constructor */
    point( double xVal, double yVal ) { x_ = xVal; y_ = yVal; }
    /** copy constructor */
    point( const point &rhs ) { x_ = rhs.x_; y_ = rhs.y_; }
    /** destructor does nothing */
    ~point() {};

    double x() { return x_; }
    void x(const double v) { x_ = v; }
    double y() { return y_; }
    void y(const double v) { y_ = v; }
  }; // class point

  /**
    Line information.

    A container for linear regression information.

    A regression line, defined by the equation y' = a + bx, where a and
    b are constants.  The constant a is the y intercept when x == 0.
    The constant b is the slope of the line.

  */
  class lineInfo
  {
  private:
    /** the slope of the line (e.g., b in y' = a + bx) */
    double slope_;
    /** the y-intercept (e.g., a, when x == 0 in y' = a + bx) */
    double intercept_;
    /** standard deviation of the points around the line */
    double stddev_;
    /** Regression error */
    double slopeError_;
    /** correlation between the x points and the y points */
    double cor_;
  public:
    /** constructor */
    lineInfo()
    {
      slope_ = 0;
      intercept_ = 0;
      cor_ = 0;
      stddev_;
      slopeError_ = 0;
    }
    ~lineInfo() {}
    /** copy constructor */
    lineInfo( const lineInfo &rhs )
    {
      slope_ = rhs.slope_;
      intercept_ = rhs.intercept_;
      cor_ = rhs.cor_;
      stddev_ = rhs.stddev_;
      slopeError_ = rhs.slopeError_;
    }
    
    double slope() { return slope_; }
    void slope( double s ) { slope_ = s; }

    double intercept() { return intercept_; }
    void intercept( double a ) { intercept_ = a; }

    double stddev() { return stddev_; }
    void stddev( double s ) { stddev_ = s; }

    double corr() { return cor_; }
    void corr( double c ) { cor_ = c; }

    double slopeErr() { return slopeError_; }
    void slopeErr(double e) { slopeError_ = e; }
  }; // class lineInfo

  double meanX( std::vector<point> &data ) const;
  double meanY( std::vector<point> &data ) const;

  /**
     Calculate the linear regression on a set of N points,
     {Xi, Yi}.
   */
  void lr( std::vector<point> &data, lineInfo &info ) const;

}; // lregress

#endif