TModeler.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 TMODELER_H
#define TMODELER_H

#include <iostream>

//ROOT
#include "Rtypes.h"
#include "TNamed.h"
//Added to make up for TMultiLayerPerceptron comment
#include "TVector.h"
//Provisoirement retir�
#ifndef WIN32
#include "TMultiLayerPerceptron.h"
#endif
//Uranie
#include "Modeler.h"
#include "TDataServer.h"
#include "UExceptions.h"

namespace URANIE
{
namespace Modeler
{
class TModeler: public TNamed
{
public:
    Bool_t _blog;                             
    Bool_t _bdrawProgressBar; 
    Bool_t _bStoreYHat; 
    URANIE::DataServer::TDataServer *_tds;     
    //TMultiLayerPerceptron* _mlp;               //!< Perceptron associated

    Int_t _nS;                            

    TString _sInputAttributes; 
    TString _sOutputAttributes; 
    TString _sIntermediateArchitecture; 
    TList *_listOfInputAttributes;             
    TList *_listOfOutputAttributes;            

    Int_t _nX;                            
    Int_t _nY;                            

    TString _sFileName; 
    TString _sFunctionName;  

    TVectorD _parameterValues;                 
    Double_t _dr2;                             
    Bool_t _bIntercept; 
public:
    //---------------------------------------------
    TModeler(URANIE::DataServer::TDataServer *tds, TString architecture,
            Option_t *option = "");
    TModeler(URANIE::DataServer::TDataServer *tds, const char *varexpinput,
            const char *varexpoutput, Option_t * option = "");

    virtual ~TModeler();

    //---------------------------------------------
    //TMultiLayerPerceptron* getANN() { return _mlp;}
    TVectorD getParameters()
    {
        return _parameterValues;
    }
    Double_t getR2() const
    {
        return _dr2;
    }
    const char * getInputName(Int_t i) const;
    const char * getOutputName(Int_t i = 0);
    Int_t getNInput();

    Bool_t getIntercept()
    {
        return _bIntercept;
    }

    void setDrawProgressBar(Bool_t bbool = kTRUE)
    {
        _bdrawProgressBar = bbool;
    }


    Bool_t getDrawProgressBar()
    {
        return _bdrawProgressBar;
    }

    //---------------------------------------------
    void train(Option_t* option = "text");

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

    virtual void exportFunction(const char* lang, const char* file = "",
            const char* name = "", Option_t *option = "");

    virtual void exportModelCplusplus(std::ofstream * sourcefile) const = 0;

    virtual void exportModelFortran(std::ofstream * sourcefile) const = 0;

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

    virtual void exportModelPython(std::ofstream * sourcefile) const = 0;

    //---------------------------------------------
    void setLog()
    {
        _blog = kTRUE;
    }
    void unsetLog()
    {
        _blog = kFALSE;
    }
    void changeLog()
    {
        _blog = _blog ? kFALSE : kTRUE;
    }
    Bool_t getLog()
    {
        return _blog;
    }
    virtual void printLog(Option_t *option = "");

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

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