Quick Links: ROOT Homepage Class Index Class Hierarchy

Source: header file source file viewVC header viewVC source

Sections: class description function members data members class charts

class TMVA::RuleFitAPI


 J Friedman's RuleFit method

Function Members (Methods)

public:

virtual	~RuleFitAPI()
void	CheckRFWorkDir()
static TClass*	Class()
const TString	GetRFWorkDir() const
void	HowtoSetupRF()
virtual TClass*	IsA() const
Bool_t	ReadModelSum()
TMVA::RuleFitAPI	RuleFitAPI(const TMVA::MethodRuleFit* rfbase, TMVA::RuleFit* rulefit, TMVA::EMsgType minType)
void	SetRFWorkDir(const char* wdir)
virtual void	ShowMembers(TMemberInspector& insp, char* parent)
virtual void	Streamer(TBuffer& b)
void	StreamerNVirtual(TBuffer& b)
void	TestRuleFit()
void	TrainRuleFit()
void	VarImp()
void	WelcomeMessage()

protected:

void	FillIntParmsDef()
void	FillRealParmsDef()
TString	GetRFName(TString name)
void	ImportSetup()
void	InitRuleFit()
Bool_t	OpenRFile(TString name, ofstream& f)
Bool_t	OpenRFile(TString name, ifstream& f)
Int_t	ReadFloat(ifstream& f, Float_t* v, Int_t n = 1) const
Int_t	ReadInt(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()
Bool_t	ReadVarNames()
Bool_t	ReadYhat()
Int_t	RunRuleFit()
void	SetRFPredict()
void	SetRFTrain()
void	SetRFVarimp()
void	SetTestParms()
void	SetTrainParms()
Bool_t	WriteAll()
Bool_t	WriteFloat(ofstream& f, const Float_t* v, Int_t n = 1)
Bool_t	WriteInt(ofstream& f, const Int_t* v, Int_t n = 1)
Bool_t	WriteIntParms()
Bool_t	WriteLx()
Bool_t	WriteProgram()
Bool_t	WriteRealParms()
Bool_t	WriteRealVarImp()
Bool_t	WriteRfOut()
Bool_t	WriteRfStatus()
Bool_t	WriteRuleFitMod()
Bool_t	WriteRuleFitSum()
Bool_t	WriteTest()
Bool_t	WriteTrain()
Bool_t	WriteVarImp()
Bool_t	WriteVarNames()
Bool_t	WriteYhat()

private:

TMVA::RuleFitAPI

RuleFitAPI()

Data Members

private:

enum ERFMode {	kRfRegress
	kRfClass
};
enum EModel {	kRfLinear
	kRfRules
	kRfBoth
};
enum ERFProgram {	kRfTrain
	kRfPredict
	kRfVarimp
};

private:

TMVA::MsgLogger	fLogger	message logger
const TMVA::MethodRuleFit*	fMethodRuleFit	parent method - set in constructor
TString	fModelType	model type string
TMVA::RuleFitAPI::IntParms	fRFIntParms	integer parameters
vector<int>	fRFLx	variable selector
TMVA::RuleFitAPI::ERFProgram	fRFProgram	what to run
TMVA::RuleFitAPI::RealParms	fRFRealParms	real parameters
vector<Float_t>	fRFVarImp	variable importances
vector<Int_t>	fRFVarImpInd	variable index
TString	fRFWorkDir	working directory
vector<Float_t>	fRFYhat	score results from test sample
TMVA::RuleFit*	fRuleFit	non const ptr to RuleFit class in MethodRuleFit

Class Charts

Inheritance Inherited Members Includes Libraries

Function documentation

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

 standard constructor

~RuleFitAPI()

 destructor

void WelcomeMessage()

 welcome message

void HowtoSetupRF()

 howto message

void InitRuleFit()

 default initialisation
   SetRFWorkDir("./rulefit");

void ImportSetup()

 import setup from MethodRuleFit

void SetRFWorkDir(const char* wdir)

 set the directory containing rf_go.exe.

void CheckRFWorkDir()

 check if the rulefit work dir is properly setup.
 it aborts (kFATAL) if not.

 Check existance of directory

void SetTrainParms()

 set the training parameters

void SetTestParms()

 set the test params

void FillRealParmsDef()

 set default real params

void FillIntParmsDef()

 set default int params

Bool_t WriteAll()

 write all files read by rf_go.exe

Bool_t WriteIntParms()

 write int params file

Bool_t WriteRealParms()

 write int params file

Bool_t WriteLx()

 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()) {

Bool_t WriteProgram()

 write command to rf_go.exe

Bool_t WriteRealVarImp()

 write the minimum importance to be considered

Bool_t WriteRfOut()

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

Bool_t WriteRfStatus()

 written by rf_go.exe; write rulefit status

Bool_t WriteRuleFitMod()

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

Bool_t WriteRuleFitSum()

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

Bool_t WriteTrain()

 write training data, columnwise

Bool_t WriteTest()

 Write test data

Bool_t WriteVarNames()

 write variable names, ascii

Bool_t WriteVarImp()

 written by rf_go.exe

Bool_t WriteYhat()

 written by rf_go.exe

Bool_t ReadYhat()

 read the score

Bool_t ReadVarImp()

 read variable importance

Bool_t ReadModelSum()

 read model from rulefit.sum

Int_t RunRuleFit()

 execute rf_go.exe

void TrainRuleFit()

void TestRuleFit()

void VarImp()

TString GetRFName(TString name)

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

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

Bool_t WriteInt(ofstream& f, const Int_t* v, Int_t n = 1)

Bool_t WriteFloat(ofstream& f, const Float_t* v, Int_t n = 1)

Int_t ReadInt(ifstream& f, Int_t* v, Int_t n = 1) const

Int_t ReadFloat(ifstream& f, Float_t* v, Int_t n = 1) const

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

const TString GetRFWorkDir() const

 Get working directory

{ return fRFWorkDir; }

void SetRFTrain()

 set rf_go.exe running mode

{ fRFProgram = kRfTrain; }

void SetRFPredict()

{ fRFProgram = kRfPredict; }

void SetRFVarimp()

{ fRFProgram = kRfVarimp; }

Bool_t ReadIntParms()

Bool_t ReadRealParms()

Bool_t ReadLx()

Bool_t ReadProgram()

Bool_t ReadRealVarImp()

Bool_t ReadRfOut()

Bool_t ReadRfStatus()

Bool_t ReadRuleFitMod()

Bool_t ReadRuleFitSum()

Bool_t ReadTrainX()

Bool_t ReadTrainY()

Bool_t ReadTrainW()

Bool_t ReadVarNames()