STK_AdditiveBSplineRegression.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 AdditiveBSplineRegression class.
 * Author:   iovleff, serge.iovleff@stkpp.org
 **/

#ifndef STK_ADDITIVEBSPLINEREGRESSION_H
#define STK_ADDITIVEBSPLINEREGRESSION_H

#include "STK_IRegression.h"
#include "STK_AdditiveBSplineCoefficients.h"

namespace STK
{

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

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

    virtual ~AdditiveBSplineRegression();

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

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

    virtual void preRun();

    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_ADDITIVEBSPLINEREGRESSION_H */