STK_BSplineRegression.h

Go to the documentation of this file.

/*--------------------------------------------------------------------*/
/*     Copyright (C) 2004-2010  Serge Iovleff

 This program is free software; you can redistribute it and/or modify
 it under the terms of the GNU Lesser General Public License as
 published by the Free Software Foundation; either version 2 of the
 License, or (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU Lesser General Public License for more details.

 You should have received a copy of the GNU Lesser General Public
 License along with this program; if not, write to the
 Free Software Foundation, Inc.,
 59 Temple Place,
 Suite 330,
 Boston, MA 02111-1307
 USA

 Contact : Serge.Iovleff@stkpp.org
 */

/*
 * Project:  stkpp::Regress
 * created on: 31 juil. 2010
 * Purpose: definition of the BsplineRegression class.
 * Author:   iovleff, serge.iovleff@stkpp.org
 **/

#ifndef STK_BSPLINEREGRESSION_H
#define STK_BSPLINEREGRESSION_H

#include "../../Arrays/include/STK_Matrix.h"

#include "STK_BSplineCoefficients.h"
#include "STK_IRegression.h"

namespace STK
{

class BSplineRegression : public IRegression<Matrix, Vector, Vector>
{
  private:
    typedef BSplineCoefficients::KnotsPosition _Kposition;

  public:
    BSplineRegression( Matrix const* p_y
                     , Vector const* p_x
                     , Integer const& nbControlPoints
                     , Integer const& degree = 3
                     , const _Kposition& position = BSplineCoefficients::uniform_
                     );

    virtual ~BSplineRegression();

    inline Integer const& degree() const
    { return degree_;}
    inline Integer const& nbControlPoints() const
    { return nbControlPoints_;}
    inline Matrix const& controlPoints() const
    { return controlPoints_; }
    inline Vector const& knots() const
    { return coefs_.knots(); }
    inline Matrix const& coefficients() const
    { return coefs_.coefficients();}

  protected:
    Integer nbControlPoints_;
    Integer degree_;
    _Kposition position_;
    BSplineCoefficients coefs_;
    Matrix controlPoints_;

    inline virtual void preRun() {coefs_.run();}

    virtual void regression();
    virtual void regression(Vector const& weights);
    virtual void prediction();
    inline virtual Integer computeNbParameter() const
    { return controlPoints_.sizeHo() * controlPoints_.sizeVe(); }

};

} // namespace STK

#endif /* STK_BSPLINEREGRESSION_H */