#ifndef ROOT_TMVA_MethodRuleFit
#define ROOT_TMVA_MethodRuleFit
#ifndef ROOT_TMVA_MethodBase
#include "TMVA/MethodBase.h"
#endif
#ifndef ROOT_TMVA_TMatrixD
#include "TMatrixD.h"
#endif
#ifndef ROOT_TMVA_TVectorD
#include "TVectorD.h"
#endif
#ifndef ROOT_TMVA_DecisionTree
#include "TMVA/DecisionTree.h"
#endif
#ifndef ROOT_TMVA_SeparationBase
#include "TMVA/SeparationBase.h"
#endif
#ifndef ROOT_TMVA_GiniIndex
#include "TMVA/GiniIndex.h"
#endif
#ifndef ROOT_TMVA_CrossEntropy
#include "TMVA/CrossEntropy.h"
#endif
#ifndef ROOT_TMVA_MisClassificationError
#include "TMVA/MisClassificationError.h"
#endif
#ifndef ROOT_TMVA_SdivSqrtSplusB
#include "TMVA/SdivSqrtSplusB.h"
#endif
#ifndef ROOT_TMVA_RULEFIT_H
#include "TMVA/RuleFit.h"
#endif
namespace TMVA {
class MethodRuleFit : public MethodBase {
public:
MethodRuleFit( TString jobName,
TString methodTitle,
DataSet& theData,
TString theOption = "",
TDirectory* theTargetDir = 0 );
MethodRuleFit( DataSet& theData,
TString theWeightFile,
TDirectory* theTargetDir = NULL );
virtual ~MethodRuleFit( void );
virtual void Train( void );
virtual void WriteWeightsToStream( ostream& o ) const;
virtual void ReadWeightsFromStream( istream& istr );
virtual Double_t GetMvaValue();
virtual void WriteMonitoringHistosToFile( void ) const;
const Ranking* CreateRanking();
const std::vector<TMVA::Event*> &GetTrainingEvents() const { return fEventSample; }
const std::vector<TMVA::DecisionTree*> &GetForest() const { return fForest; }
const Int_t GetNTrees() const { return fNTrees; }
const Double_t GetSampleFraction() const { return fSampleFraction; }
const SeparationBase *GetSeparationBase() const { return fSepType; }
const Int_t GetNCuts() const { return fNCuts; }
protected:
void InitRuleFit( void );
void InitEventSample( void );
void InitMonitorNtuple();
void BuildTree( DecisionTree *dt, std::vector< Event *> & el );
void MakeForest();
std::vector< Event *> fEventSample;
std::vector<DecisionTree *> fForest;
Int_t fNTrees;
Double_t fSampleFraction;
SeparationBase *fSepType;
Int_t fNodeMinEvents;
Int_t fNCuts;
RuleFit fRuleFit;
private:
virtual void DeclareOptions();
virtual void ProcessOptions();
Double_t fSignalFraction;
TTree *fMonitorNtuple;
Double_t fNTImportance;
Double_t fNTCoefficient;
Double_t fNTSupport;
Int_t fNTNcuts;
Double_t fNTPtag;
Double_t fNTPss;
Double_t fNTPsb;
Double_t fNTPbs;
Double_t fNTPbb;
Double_t fNTSSB;
Int_t fNTType;
Double_t fGDTau;
Double_t fGDPathStep;
Int_t fGDNPathSteps;
Double_t fGDErrNsigma;
Double_t fMinimp;
TString fSepTypeS;
TString fModelTypeS;
Double_t fRuleMaxDist;
ClassDef(MethodRuleFit,0)
};
}
#endif // MethodRuleFit_H
ROOT page - Class index - Class Hierarchy - Top of the page
This page has been automatically generated. If you have any comments or suggestions about the page layout send a mail to ROOT support, or contact the developers with any questions or problems regarding ROOT.