STK_BernoulliModel.h

Go to the documentation of this file.

/*--------------------------------------------------------------------*/
/*     Copyright (C) 2004-2016  Serge Iovleff, Université Lille 1, Inria
 
    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 : S..._Dot_I..._At_stkpp_Dot_org (see copyright for ...)
*/
 
/*
 * Project:  stkpp::Model
 * created on: 22 juil. 2011
 * Purpose: define the class IUnivStatModel.
 * Author:   iovleff, S..._Dot_I..._At_stkpp_Dot_org (see copyright for ...)
 *
 **/
 
#ifndef STK_BERNOULLIMODEL_H
#define STK_BERNOULLIMODEL_H
 
#include <cmath>
 
#include "STK_IUnivStatModel.h"
#include "STatistiK/include/STK_Law_Bernoulli.h"
#include "Arrays/include/STK_CArrayVector.h"
 
namespace STK
{
 
template <class Array, class WColVector>
class BernoulliModel: public IUnivStatModel<Array, WColVector, Law::Bernoulli>
{
  using IUnivStatModel<Array, WColVector, Law::Bernoulli>::p_data;
 
  public:
    typedef typename Array::Type Type;
    typedef IUnivStatModel<Array, WColVector, Law::Bernoulli> Base;
    BernoulliModel(): Base() {}
    BernoulliModel(Array const& data): Base(data) {}
    BernoulliModel(Array const* p_data): Base(p_data) {}
    BernoulliModel(BernoulliModel const& model): Base(model) {}
    BernoulliModel* clone() const { return new BernoulliModel(*this);}
    virtual ~BernoulliModel() {}
    virtual int computeNbFreeParameters() const { return 1;}
 
  protected:
    virtual void computeParameters()
    {
        Real sum=0.;
        int nbObs=p_data()->size();
        for (int i=p_data()->begin(); i<=p_data()->lastIdx(); ++i)
        { (p_data()->elt(i) == binaryNA_) ? --nbObs : sum += p_data()->elt(i);}
        if (nbObs != 0) { this->law_.setProb(sum/nbObs);}
                  else  { this->law_.setProb(0.);}
    }
    virtual void computeParameters(WColVector const& weights)
    {
        Real sum=0.;
        int nbObs=p_data()->size();
        for (int i=p_data()->begin(); i<=p_data()->lastIdx(); ++i)
        { (p_data()->elt(i) == binaryNA_) ? --nbObs : sum += weights[i]*p_data()->elt(i);}
        if (nbObs != 0) { this->law_.setProb(sum/nbObs);}
                  else { this->law_.setProb(0.);}
    }
};
 
} // namespace STK
 
#endif /* STK_BERNOULLIMODEL_H */