polynomialRegression.hpp

Go to the documentation of this file.

#ifndef JULIAN_POLYNOMIALREGRESSION_HPP
#define JULIAN_POLYNOMIALREGRESSION_HPP

#include <vector>
#include <iostream>
#include <mathematics/regressions/regression.hpp>

namespace julian {

  class PolynomialRegression : public DeeplyCopyableRegression<PolynomialRegression> {
  public:
    PolynomialRegression(): order_(2) {};

    PolynomialRegression(int order): order_(order) {};

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

    std::vector<double> getCoefficient() const;
    
    double operator()(double) const;

  private:
    int order_; 
    std::vector<double> coefs_; 
  };
}

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

#endif