Logo ROOT   6.10/09
Reference Guide
List of all members | Classes | Public Member Functions | Protected Types | Protected Member Functions | Private Member Functions | Private Attributes | List of all members
TMVA::RuleFitAPI Class Reference

J Friedman's RuleFit method.

Definition at line 50 of file RuleFitAPI.h.

Classes

struct  IntParms
 
struct  RealParms
 

Public Member Functions

 RuleFitAPI (const TMVA::MethodRuleFit *rfbase, TMVA::RuleFit *rulefit, EMsgType minType)
 
virtual ~RuleFitAPI ()
 destructor More...
 
void CheckRFWorkDir ()
 check if the rulefit work dir is properly setup. More...
 
const TString GetRFWorkDir () const
 
void HowtoSetupRF ()
 howto message More...
 
Bool_t ReadModelSum ()
 read model from rulefit.sum More...
 
void SetRFWorkDir (const char *wdir)
 set the directory containing rf_go.exe. More...
 
void TestRuleFit ()
 
void TrainRuleFit ()
 
void VarImp ()
 
void WelcomeMessage ()
 welcome message More...
 

Protected Types

enum  EModel { kRfLinear =0, kRfRules =1, kRfBoth =2 }
 
enum  ERFMode { kRfRegress =1, kRfClass =2 }
 
enum  ERFProgram { kRfTrain =0, kRfPredict, kRfVarimp }
 

Protected Member Functions

void FillIntParmsDef ()
 set default int params More...
 
void FillRealParmsDef ()
 set default real params More...
 
TString GetRFName (TString name)
 
void ImportSetup ()
 import setup from MethodRuleFit More...
 
void InitRuleFit ()
 default initialisation SetRFWorkDir("./rulefit"); More...
 
Bool_t OpenRFile (TString name, std::ofstream &f)
 
Bool_t OpenRFile (TString name, std::ifstream &f)
 
Int_t ReadFloat (std::ifstream &f, Float_t *v, Int_t n=1) const
 
Int_t ReadInt (std::ifstream &f, Int_t *v, Int_t n=1) const
 
Bool_t ReadIntParms ()
 
Bool_t ReadLx ()
 
Bool_t ReadProgram ()
 
Bool_t ReadRealParms ()
 
Bool_t ReadRealVarImp ()
 
Bool_t ReadRfOut ()
 
Bool_t ReadRfStatus ()
 
Bool_t ReadRuleFitMod ()
 
Bool_t ReadRuleFitSum ()
 
Bool_t ReadTrainW ()
 
Bool_t ReadTrainX ()
 
Bool_t ReadTrainY ()
 
Bool_t ReadVarImp ()
 read variable importance More...
 
Bool_t ReadVarNames ()
 
Bool_t ReadYhat ()
 read the score More...
 
Int_t RunRuleFit ()
 execute rf_go.exe More...
 
void SetRFPredict ()
 
void SetRFTrain ()
 
void SetRFVarimp ()
 
void SetTestParms ()
 set the test params More...
 
void SetTrainParms ()
 set the training parameters More...
 
Bool_t WriteAll ()
 write all files read by rf_go.exe More...
 
Bool_t WriteFloat (std::ofstream &f, const Float_t *v, Int_t n=1)
 
Bool_t WriteInt (std::ofstream &f, const Int_t *v, Int_t n=1)
 
Bool_t WriteIntParms ()
 write int params file More...
 
Bool_t WriteLx ()
 Save input variable mask. More...
 
Bool_t WriteProgram ()
 write command to rf_go.exe More...
 
Bool_t WriteRealParms ()
 write int params file More...
 
Bool_t WriteRealVarImp ()
 write the minimum importance to be considered More...
 
Bool_t WriteRfOut ()
 written by rf_go.exe; write rulefit output (rfout) More...
 
Bool_t WriteRfStatus ()
 written by rf_go.exe; write rulefit status More...
 
Bool_t WriteRuleFitMod ()
 written by rf_go.exe (NOTE:Format unknown!) More...
 
Bool_t WriteRuleFitSum ()
 written by rf_go.exe (NOTE: format unknown!) More...
 
Bool_t WriteTest ()
 Write test data. More...
 
Bool_t WriteTrain ()
 write training data, column wise More...
 
Bool_t WriteVarImp ()
 
Bool_t WriteVarNames ()
 write variable names, ascii More...
 
Bool_t WriteYhat ()
 written by rf_go.exe More...
 

Private Member Functions

 RuleFitAPI ()
 

Private Attributes

MsgLogger fLogger
 
const MethodRuleFitfMethodRuleFit
 
TString fModelType
 
IntParms fRFIntParms
 
std::vector< int > fRFLx
 
ERFProgram fRFProgram
 
RealParms fRFRealParms
 
std::vector< Float_tfRFVarImp
 
std::vector< Int_tfRFVarImpInd
 
TString fRFWorkDir
 
std::vector< Float_tfRFYhat
 
RuleFitfRuleFit
 

#include <TMVA/RuleFitAPI.h>

Member Enumeration Documentation

◆ EModel

enum TMVA::RuleFitAPI::EModel
protected
Enumerator
kRfLinear 
kRfRules 
kRfBoth 

Definition at line 84 of file RuleFitAPI.h.

◆ ERFMode

enum TMVA::RuleFitAPI::ERFMode
protected
Enumerator
kRfRegress 
kRfClass 

Definition at line 83 of file RuleFitAPI.h.

◆ ERFProgram

enum TMVA::RuleFitAPI::ERFProgram
protected
Enumerator
kRfTrain 
kRfPredict 
kRfVarimp 

Definition at line 85 of file RuleFitAPI.h.

Constructor & Destructor Documentation

◆ RuleFitAPI() [1/2]

TMVA::RuleFitAPI::RuleFitAPI ( const TMVA::MethodRuleFit rfbase,
TMVA::RuleFit rulefit,
EMsgType  minType = kINFO 
)

Definition at line 50 of file RuleFitAPI.cxx.

◆ ~RuleFitAPI()

TMVA::RuleFitAPI::~RuleFitAPI ( )
virtual

destructor

Definition at line 71 of file RuleFitAPI.cxx.

◆ RuleFitAPI() [2/2]

TMVA::RuleFitAPI::RuleFitAPI ( )
private

Member Function Documentation

◆ CheckRFWorkDir()

void TMVA::RuleFitAPI::CheckRFWorkDir ( )

check if the rulefit work dir is properly setup.

it aborts (kFATAL) if not.

Check existence of directory

Definition at line 168 of file RuleFitAPI.cxx.

◆ FillIntParmsDef()

void TMVA::RuleFitAPI::FillIntParmsDef ( )
protected

set default int params

Definition at line 230 of file RuleFitAPI.cxx.

◆ FillRealParmsDef()

void TMVA::RuleFitAPI::FillRealParmsDef ( )
protected

set default real params

Definition at line 215 of file RuleFitAPI.cxx.

◆ GetRFName()

TString TMVA::RuleFitAPI::GetRFName ( TString  name)
inlineprotected

Definition at line 230 of file RuleFitAPI.h.

◆ GetRFWorkDir()

const TString TMVA::RuleFitAPI::GetRFWorkDir ( ) const
inline

Definition at line 79 of file RuleFitAPI.h.

◆ HowtoSetupRF()

void TMVA::RuleFitAPI::HowtoSetupRF ( )

howto message

Definition at line 94 of file RuleFitAPI.cxx.

◆ ImportSetup()

void TMVA::RuleFitAPI::ImportSetup ( )
protected

import setup from MethodRuleFit

Definition at line 134 of file RuleFitAPI.cxx.

◆ InitRuleFit()

void TMVA::RuleFitAPI::InitRuleFit ( )
protected

default initialisation SetRFWorkDir("./rulefit");

Definition at line 124 of file RuleFitAPI.cxx.

◆ OpenRFile() [1/2]

Bool_t TMVA::RuleFitAPI::OpenRFile ( TString  name,
std::ofstream &  f 
)
inlineprotected

Definition at line 237 of file RuleFitAPI.h.

◆ OpenRFile() [2/2]

Bool_t TMVA::RuleFitAPI::OpenRFile ( TString  name,
std::ifstream &  f 
)
inlineprotected

Definition at line 251 of file RuleFitAPI.h.

◆ ReadFloat()

Int_t TMVA::RuleFitAPI::ReadFloat ( std::ifstream &  f,
Float_t v,
Int_t  n = 1 
) const
inlineprotected

Definition at line 290 of file RuleFitAPI.h.

◆ ReadInt()

Int_t TMVA::RuleFitAPI::ReadInt ( std::ifstream &  f,
Int_t v,
Int_t  n = 1 
) const
inlineprotected

Definition at line 281 of file RuleFitAPI.h.

◆ ReadIntParms()

Bool_t TMVA::RuleFitAPI::ReadIntParms ( )
protected

◆ ReadLx()

Bool_t TMVA::RuleFitAPI::ReadLx ( )
protected

◆ ReadModelSum()

Bool_t TMVA::RuleFitAPI::ReadModelSum ( )

read model from rulefit.sum

Definition at line 546 of file RuleFitAPI.cxx.

◆ ReadProgram()

Bool_t TMVA::RuleFitAPI::ReadProgram ( )
protected

◆ ReadRealParms()

Bool_t TMVA::RuleFitAPI::ReadRealParms ( )
protected

◆ ReadRealVarImp()

Bool_t TMVA::RuleFitAPI::ReadRealVarImp ( )
protected

◆ ReadRfOut()

Bool_t TMVA::RuleFitAPI::ReadRfOut ( )
protected

◆ ReadRfStatus()

Bool_t TMVA::RuleFitAPI::ReadRfStatus ( )
protected

◆ ReadRuleFitMod()

Bool_t TMVA::RuleFitAPI::ReadRuleFitMod ( )
protected

◆ ReadRuleFitSum()

Bool_t TMVA::RuleFitAPI::ReadRuleFitSum ( )
protected

◆ ReadTrainW()

Bool_t TMVA::RuleFitAPI::ReadTrainW ( )
protected

◆ ReadTrainX()

Bool_t TMVA::RuleFitAPI::ReadTrainX ( )
protected

◆ ReadTrainY()

Bool_t TMVA::RuleFitAPI::ReadTrainY ( )
protected

◆ ReadVarImp()

Bool_t TMVA::RuleFitAPI::ReadVarImp ( )
protected

read variable importance

Definition at line 509 of file RuleFitAPI.cxx.

◆ ReadVarNames()

Bool_t TMVA::RuleFitAPI::ReadVarNames ( )
protected

◆ ReadYhat()

Bool_t TMVA::RuleFitAPI::ReadYhat ( )
protected

read the score

Definition at line 484 of file RuleFitAPI.cxx.

◆ RunRuleFit()

Int_t TMVA::RuleFitAPI::RunRuleFit ( )
protected

execute rf_go.exe

Definition at line 774 of file RuleFitAPI.cxx.

◆ SetRFPredict()

void TMVA::RuleFitAPI::SetRFPredict ( )
inlineprotected

Definition at line 128 of file RuleFitAPI.h.

◆ SetRFTrain()

void TMVA::RuleFitAPI::SetRFTrain ( )
inlineprotected

Definition at line 127 of file RuleFitAPI.h.

◆ SetRFVarimp()

void TMVA::RuleFitAPI::SetRFVarimp ( )
inlineprotected

Definition at line 129 of file RuleFitAPI.h.

◆ SetRFWorkDir()

void TMVA::RuleFitAPI::SetRFWorkDir ( const char *  wdir)

set the directory containing rf_go.exe.

Definition at line 157 of file RuleFitAPI.cxx.

◆ SetTestParms()

void TMVA::RuleFitAPI::SetTestParms ( )
protected

set the test params

Definition at line 203 of file RuleFitAPI.cxx.

◆ SetTrainParms()

void TMVA::RuleFitAPI::SetTrainParms ( )
protected

set the training parameters

Definition at line 190 of file RuleFitAPI.cxx.

◆ TestRuleFit()

void TMVA::RuleFitAPI::TestRuleFit ( )
inline

Definition at line 210 of file RuleFitAPI.h.

◆ TrainRuleFit()

void TMVA::RuleFitAPI::TrainRuleFit ( )
inline

Definition at line 201 of file RuleFitAPI.h.

◆ VarImp()

void TMVA::RuleFitAPI::VarImp ( )
inline

Definition at line 220 of file RuleFitAPI.h.

◆ WelcomeMessage()

void TMVA::RuleFitAPI::WelcomeMessage ( )

welcome message

Definition at line 78 of file RuleFitAPI.cxx.

◆ WriteAll()

Bool_t TMVA::RuleFitAPI::WriteAll ( )
protected

write all files read by rf_go.exe

Definition at line 250 of file RuleFitAPI.cxx.

◆ WriteFloat()

Bool_t TMVA::RuleFitAPI::WriteFloat ( std::ofstream &  f,
const Float_t v,
Int_t  n = 1 
)
inlineprotected

Definition at line 273 of file RuleFitAPI.h.

◆ WriteInt()

Bool_t TMVA::RuleFitAPI::WriteInt ( std::ofstream &  f,
const Int_t v,
Int_t  n = 1 
)
inlineprotected

Definition at line 265 of file RuleFitAPI.h.

◆ WriteIntParms()

Bool_t TMVA::RuleFitAPI::WriteIntParms ( )
protected

write int params file

Definition at line 266 of file RuleFitAPI.cxx.

◆ WriteLx()

Bool_t TMVA::RuleFitAPI::WriteLx ( )
protected

Save input variable mask.

If the lx vector size is not the same as inputVars, resize it and fill it with 1 NOTE: Always set all to 1 if (fRFLx.size() != m_inputVars->size()) {

Definition at line 293 of file RuleFitAPI.cxx.

◆ WriteProgram()

Bool_t TMVA::RuleFitAPI::WriteProgram ( )
protected

write command to rf_go.exe

Definition at line 307 of file RuleFitAPI.cxx.

◆ WriteRealParms()

Bool_t TMVA::RuleFitAPI::WriteRealParms ( )
protected

write int params file

Definition at line 277 of file RuleFitAPI.cxx.

◆ WriteRealVarImp()

Bool_t TMVA::RuleFitAPI::WriteRealVarImp ( )
protected

write the minimum importance to be considered

Definition at line 335 of file RuleFitAPI.cxx.

◆ WriteRfOut()

Bool_t TMVA::RuleFitAPI::WriteRfOut ( )
protected

written by rf_go.exe; write rulefit output (rfout)

Definition at line 349 of file RuleFitAPI.cxx.

◆ WriteRfStatus()

Bool_t TMVA::RuleFitAPI::WriteRfStatus ( )
protected

written by rf_go.exe; write rulefit status

Definition at line 358 of file RuleFitAPI.cxx.

◆ WriteRuleFitMod()

Bool_t TMVA::RuleFitAPI::WriteRuleFitMod ( )
protected

written by rf_go.exe (NOTE:Format unknown!)

Definition at line 367 of file RuleFitAPI.cxx.

◆ WriteRuleFitSum()

Bool_t TMVA::RuleFitAPI::WriteRuleFitSum ( )
protected

written by rf_go.exe (NOTE: format unknown!)

Definition at line 376 of file RuleFitAPI.cxx.

◆ WriteTest()

Bool_t TMVA::RuleFitAPI::WriteTest ( )
protected

Write test data.

Definition at line 420 of file RuleFitAPI.cxx.

◆ WriteTrain()

Bool_t TMVA::RuleFitAPI::WriteTrain ( )
protected

write training data, column wise

Definition at line 385 of file RuleFitAPI.cxx.

◆ WriteVarImp()

Bool_t TMVA::RuleFitAPI::WriteVarImp ( )
protected

Definition at line 464 of file RuleFitAPI.cxx.

◆ WriteVarNames()

Bool_t TMVA::RuleFitAPI::WriteVarNames ( )
protected

write variable names, ascii

Definition at line 452 of file RuleFitAPI.cxx.

◆ WriteYhat()

Bool_t TMVA::RuleFitAPI::WriteYhat ( )
protected

written by rf_go.exe

Definition at line 475 of file RuleFitAPI.cxx.

Member Data Documentation

◆ fLogger

MsgLogger TMVA::RuleFitAPI::fLogger
mutableprivate

Definition at line 192 of file RuleFitAPI.h.

◆ fMethodRuleFit

const MethodRuleFit* TMVA::RuleFitAPI::fMethodRuleFit
private

Definition at line 179 of file RuleFitAPI.h.

◆ fModelType

TString TMVA::RuleFitAPI::fModelType
private

Definition at line 190 of file RuleFitAPI.h.

◆ fRFIntParms

IntParms TMVA::RuleFitAPI::fRFIntParms
private

Definition at line 186 of file RuleFitAPI.h.

◆ fRFLx

std::vector<int> TMVA::RuleFitAPI::fRFLx
private

Definition at line 188 of file RuleFitAPI.h.

◆ fRFProgram

ERFProgram TMVA::RuleFitAPI::fRFProgram
private

Definition at line 189 of file RuleFitAPI.h.

◆ fRFRealParms

RealParms TMVA::RuleFitAPI::fRFRealParms
private

Definition at line 187 of file RuleFitAPI.h.

◆ fRFVarImp

std::vector<Float_t> TMVA::RuleFitAPI::fRFVarImp
private

Definition at line 183 of file RuleFitAPI.h.

◆ fRFVarImpInd

std::vector<Int_t> TMVA::RuleFitAPI::fRFVarImpInd
private

Definition at line 184 of file RuleFitAPI.h.

◆ fRFWorkDir

TString TMVA::RuleFitAPI::fRFWorkDir
private

Definition at line 185 of file RuleFitAPI.h.

◆ fRFYhat

std::vector<Float_t> TMVA::RuleFitAPI::fRFYhat
private

Definition at line 182 of file RuleFitAPI.h.

◆ fRuleFit

RuleFit* TMVA::RuleFitAPI::fRuleFit
private

Definition at line 180 of file RuleFitAPI.h.


The documentation for this class was generated from the following files: