simpleLinearRegression.hpp

Go to the documentation of this file.

#ifndef JULIAN_SIMPLELINEARREGRESSION_HPP
#define JULIAN_SIMPLELINEARREGRESSION_HPP

#include <mathematics/regressions/regression.hpp>
#include <iostream>
#include <vector>
#include <string>
#include <gsl/gsl_fit.h>

namespace julian {

  class SimpleLinearRegression : public DeeplyCopyableRegression<SimpleLinearRegression> {
  public:
    SimpleLinearRegression(): chi_sq_(0.0), cov00_(0.0), cov01_(0.0), cov11_(0.0), c0_(0.0), c1_(0.0) {};

    void estimate(const std::vector<double>& x,const std::vector<double>& y);
    void estimate(const std::vector<double>& x,const std::vector<double>& y,const std::vector<double>& w);
    
    double getSlope();
    double getIntercept();
    std::vector<double> getCoefficient() const;
    double operator()(double) const;

  private:
    double chi_sq_; 
    double cov00_;  
    double cov01_;  
    double cov11_;  
    double c0_;     
    double c1_;     
  };
}  //  namespace julian

std::ostream& operator<<(std::ostream& s, julian::SimpleLinearRegression& r);

#endif