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