// @(#)root/tmva $Id: TMVA_MethodRuleFit.h,v 1.1 2006/05/08 12:46:31 brun Exp $ // Author: Andreas Hoecker, Fredrik Tegenfeldt, Helge Voss, Kai Voss /********************************************************************************** * Project: TMVA - a Root-integrated toolkit for multivariate data analysis * * Package: TMVA * * Class : TMVA_MethodRuleFit * * * * Description: * * Friedman's RuleFit method -- not yet implemented -- dummy class -- * * * * Authors (alphabetical): * * Andreas Hoecker - CERN, Switzerland * * Fredrik Tegenfeldt - Iowa State U., USA * * Helge Voss - MPI-KP Heidelberg, Ger. * * Kai Voss - U. of Victoria, Canada * * * * Copyright (c) 2005: * * CERN, Switzerland, * * U. of Victoria, Canada, * * MPI-KP Heidelberg, Germany * * LAPP, Annecy, France * * * * Redistribution and use in source and binary forms, with or without * * modification, are permitted according to the terms listed in LICENSE * * (http://mva.sourceforge.net/license.txt) * * * * File and Version Information: * * $Id: TMVA_MethodRuleFit.h,v 1.1 2006/05/08 12:46:31 brun Exp $ **********************************************************************************/ #ifndef ROOT_TMVA_MethodRuleFit #define ROOT_TMVA_MethodRuleFit ////////////////////////////////////////////////////////////////////////// // // // TMVA_MethodRuleFit // // // // Friedman's RuleFit method -- not yet implemented -- dummy class -- // // // ////////////////////////////////////////////////////////////////////////// #ifndef ROOT_TMVA_MethodBase #include "TMVA_MethodBase.h" #endif #ifndef ROOT_TMatrixD #include "TMatrixD.h" #endif #ifndef ROOT_TVectorD #include "TVectorD.h" #endif class TMVA_MethodRuleFit : public TMVA_MethodBase { public: TMVA_MethodRuleFit( TString jobName, vector* theVariables, TTree* theTree = 0, TString theOption = "", TDirectory* theTargetDir = 0 ); TMVA_MethodRuleFit( vector *theVariables, TString theWeightFile, TDirectory* theTargetDir = NULL ); virtual ~TMVA_MethodRuleFit( void ); // training method virtual void Train( void ); // write weights to file virtual void WriteWeightsToFile( void ); // read weights from file virtual void ReadWeightsFromFile( void ); // calculate the MVA value virtual Double_t GetMvaValue(TMVA_Event *e); // write method specific histos to target file virtual void WriteHistosToFile( void ) ; void InitRuleFit( void ); protected: private: ClassDef(TMVA_MethodRuleFit,0) //Friedman's RuleFit method }; #endif // TMVA_MethodRuleFit_H