TLinearRegression.h

Go to the documentation of this file.

// Copyright (C) 2013-2024 CEA/DES
//
// 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 3 of the License, or 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, see <http://www.gnu.org/licenses/>.
// $Id$
// $Author$
// $Date$
// $Revision$
// $State$

#ifndef TLINEARREGRESSION_H
#define TLINEARREGRESSION_H
#include "TModeler.h"

#include <iostream>

namespace URANIE
{
namespace Modeler
{
class TLinearRegression: public TModeler
{

private:
    TMatrixD computeH(TMatrixD A);
    TMatrixD computeCov(TMatrixD A);
public:
    TVectorD computeHii(TMatrixD A);
public:
    Double_t _dr2a;                             
    Double_t _dq2;                              
    Double_t _dpress;                              
    TMatrixD _mParameterCov;
    
    // Operations
public:
    //---------------------------------------------
    /*
     \param tds (URANIE::DataServer::TDataServer *) the dataserver
     \param varexpinput(const char*)[""] The list of input attributes to pass to the function separated by the caracter ":"
     \param varexpoutput (const char *)[""]  The list of output attributes separated by the caracter ":"
     \param 
     */
    TLinearRegression(URANIE::DataServer::TDataServer *tds,
            TString architecture, Option_t *option = "");
    TLinearRegression(URANIE::DataServer::TDataServer *tds,
            const char *varexpinput, const char *varexpoutput,
            Option_t * option = "");
#ifdef  WITH_LIBXML2       
    TLinearRegression (URANIE::DataServer::TDataServer *tds,
            int id,
            const char* pmmlfile,
            const char* LRname,
            Option_t * option="");
#endif                         
    virtual ~TLinearRegression();

    //---------------------------------------------
    Double_t getR2Adjusted() const
    {
        return _dr2a;
    }
    Double_t getPress() const
    {
        return _dpress;
    }
    Double_t getQ2() const
    {
        return _dq2;
    }
    TMatrixD getParametersCovariance() const
    {
        return _mParameterCov;
    }
    //---------------------------------------------
    void estimate(Option_t * option = "");

    //---------------------------------------------

    void exportModelCplusplus(std::ofstream * sourcefile) const;

    void exportModelFortran(std::ofstream * sourcefile) const;

    void exportModelPMML(const char* file = "", const char* name = "",
            Option_t *option = "") const;
            
    

    void exportModelPython(std::ofstream * sourcefile) const
    {
        UNUSED(sourcefile);
        std::cerr
                << "   --- Method TLinearRegression::exportModelPython(ofstream * sourcefile) not yet implemented"
                << endl;
    };        

    //---------------------------------------------
    virtual void printLog(Option_t *option = "");

    ClassDef(URANIE::Modeler::TLinearRegression, ID_MODELER)
    //Classe de
};

} // Fin du namespace Modeler
} // Fin du namespace URANIE
#endif