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;
 
 
    // ------------------------- Data ----------------------------------------
 
  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)