SimpleLinearRegression.h

 
#pragma once
 
#if defined(SimpleLinearRegression_RECURSES)
#error Recursive header files inclusion detected in SimpleLinearRegression.h
#else // defined(SimpleLinearRegression_RECURSES)
#define SimpleLinearRegression_RECURSES
 
#if !defined SimpleLinearRegression_h
#define SimpleLinearRegression_h
 
 
#include <iostream>
#include <vector>
 
 
namespace DGtal
{
 
  // class SimpleLinearRegression
 
  class SimpleLinearRegression
  {
 
    // ----------------------- Standard services ------------------------------
  public:
    
    ~SimpleLinearRegression();
    
    SimpleLinearRegression( double eps_zero = 1e-8 );
 
    void clear();
 
    template <class XIterator, class YIterator>
    void addSamples( XIterator begin_x, XIterator end_x, YIterator begin_y );
 
    void addSample( const double x, const double y );
 
    bool computeRegression();
 
    double slope() const;
 
 
    double intercept() const;
 
    double estimateY( double x ) const;
 
    double estimateVariance() const;
 
    std::pair<double,double> trustIntervalForY( const double x,
                                                const double a ) const;
 
 
    void setEpsilonZero(const double aEpsilonZero)
    {
      myEpsilonZero = aEpsilonZero;
    }
 
    double epsilonZero() const
    {
      return myEpsilonZero;
    }
 
    double size() const
    {
      return myN;
    }
 
    // ----------------------- Interface --------------------------------------
  public:
 
    void selfDisplay( std::ostream & that_stream ) const;
    bool isValid() const;
 
 
    // ------------------------- Datas ----------------------------------------
 
  private:
 
    double myEpsilonZero;
 
    unsigned int myN;
 
    std::vector<double> myY;
    
    std::vector<double> myX;
    
    double myB[ 2 ];
 
    //Internal quantities.
    std::vector<double> myU;
    double mySumX;
    double mySumX2;
    double mySumY;
    double mySumXY;
    double myD;
    double myNormU2;
 
    // ------------------------- Hidden services ------------------------------
  protected:
 
  private:
 
    SimpleLinearRegression( const SimpleLinearRegression & other );
 
     SimpleLinearRegression & operator=( const SimpleLinearRegression & other );
 
    // ------------------------- Internals ------------------------------------
  private:
 
 
  };
 
  std::ostream&
  operator<<( std::ostream & that_stream,
              const SimpleLinearRegression & that_object_to_display );
 
 
} // namespace ImaGene
 
 
// Includes inline functions/methods if necessary.
#include "DGtal/math/SimpleLinearRegression.ih"
 
//                                                                           //
 
#endif // !defined SimpleLinearRegression_h
 
#undef SimpleLinearRegression_RECURSES
#endif // else defined(SimpleLinearRegression_RECURSES)