TKernel.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 TKERNEL_H
#define TKERNEL_H
 
#include "TMatrixDSymfwd.h"
 
#include "TModeler.h"
 
namespace URANIE
{
namespace Modeler
{
class TKernel: public TModeler
{
public:
    enum EKernelType
    {
        kUnknown,
        kLinear,
        kPoly,
        kSigmoid,
        kRBF,
        kLaplace,
        kSpline1,
        kBernoulli1
    };
public:
    //---------------------------------------------
    TKernel(URANIE::DataServer::TDataServer *tds, TString architecture, Option_t *option = "");
 
    TKernel(URANIE::DataServer::TDataServer *tds, const char *varexpinput,
            const char *varexpoutput, Option_t * option = "");
    virtual ~TKernel();
 
    //---------------------------------------------
 
    void setKernel(EKernelType nKernel);
    void setKernel(EKernelType nKernel, Double_t dparam1);
 
    void setKernel(EKernelType nKernel, Double_t dparam1, Double_t dparam2);
 
    void setKernel(EKernelType nKernel, Double_t dparam1, Double_t dparam2,
            Double_t dparam3);
    EKernelType getKernel()
    {
        return _nKernelType;
    }
    Double_t getParameter1()
    {
        return _dParameter1;
    }
    Double_t getParameter2()
    {
        return _dParameter2;
    }
    void setLambda(Double_t dlambda)
    {
        if (dlambda > 0)
            _dLambda = dlambda;
    }
    Double_t getLambda()
    {
        return _dLambda;
    }
    TMatrixDSym getMatrixK() { return _matK;}
 
    //---------------------------------------------
    void estimate(Option_t * option = "");
 
    void fillData();
 
    void computeK(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
    {
        TString soption = option;
        soption.ToLower();
        UNUSED(file,name);
        std::cerr
                << "   --- Method TKernel::exportModelPMML(std::ofstream * sourcefile) not yet implemented"
                << endl;
    };
    
 
    void exportModelPython(std::ofstream * sourcefile) const
    {
        UNUSED(sourcefile);
        std::cerr
                << "   --- Method TKernel::exportModelPython(std::ofstream * sourcefile) not yet implemented"
                << endl;
    };
 
    //---------------------------------------------
    void printLog(Option_t *option = "");
 
    //---------------------------------------------
private:
    EKernelType _nKernelType;        
    Double_t _dParameter1;           
    Double_t _dParameter2;           
    Double_t _dParameter3;           
    Int_t _nParameter;               
 
    TMatrixD _matX;                 
    TVectorD _vecY;                 
 
    TMatrixDSym _matK;              
    Double_t _dLambda;               
 
    ClassDef(URANIE::Modeler::TKernel, ID_MODELER)
};
} // End Of namespace Modeler
} // End Of namespace URANIE
#endif