11#ifndef ROOT_TMVA_Envelope
12#define ROOT_TMVA_Envelope
112 return fDataLoader->GetDataSetInfo().GetDataSetManager();
#define ClassDef(name, id)
This class provides a simple interface to execute the same task multiple times in parallel,...
Describe directory structure in memory.
A ROOT file is a suite of consecutive data records (TKey instances) with a well defined format.
Class that contains all the data information.
Class that contains all the data information.
Abstract base class for all high level ml algorithms, you can book ml methods like BDT,...
Bool_t HasMethod(TString methodname, TString methodtitle)
function to check methods booked
~Envelope()
Default destructor.
TStopwatch fTimer
! timer to measure the time.
Bool_t IsModelPersistence()
Method to see if the algorithm model is saved in xml or serialized files.
TDirectory * RootBaseDir()
Utility method to get base dir directory from current file.
std::shared_ptr< TFile > fFile
! file to save the results
DataLoader * GetDataLoader()
Method to get the pointer to TMVA::DataLoader object.
Bool_t fModelPersistence
! flag to save the trained model
DataInputHandler & GetDataLoaderDataInput()
Utility method to get TMVA::DataInputHandler reference from the DataLoader.
Bool_t IsSilentFile()
Method to see if a file is available to save results.
void SetDataLoader(DataLoader *dalaloader)
Method to set the pointer to TMVA::DataLoader object.
virtual void BookMethod(TString methodname, TString methodtitle, TString options="")
Method to book the machine learning method to perform the algorithm.
std::vector< OptionMap > fMethods
! Booked method information
TProcPool fWorkers
! procpool object
void SetVerbose(Bool_t status)
Method enable print extra information in the algorithms.
void SetFile(TFile *file)
Method to set the pointer to TFile object, with a writable file.
Bool_t IsVerbose()
Method to see if the algorithm should print extra information.
virtual void Evaluate()=0
Virtual method to be implemented with your algorithm.
Bool_t fVerbose
! flag for extra information
void SetModelPersistence(Bool_t status=kTRUE)
Method enable model persistence, then algorithms model is saved in xml or serialized files.
std::shared_ptr< DataLoader > fDataLoader
! data
virtual void ParseOptions()
Method to parse the internal option string.
Bool_t fSilentFile
! if true dont produce file output
DataSetManager * GetDataLoaderDataSetManager()
Utility method to get TMVA::DataSetManager pointer from the DataLoader.
TFile * GetFile()
Method to get the pointer to TFile object.
std::vector< OptionMap > & GetMethods()
Method get the Booked methods in a option map object.
TString fTransformations
! List of transformations to test
UInt_t fJobs
! number of jobs to run some high level algorithm in parallel
Envelope(const TString &name, DataLoader *dataloader=nullptr, TFile *file=nullptr, const TString options="")
Constructor for the initialization of Envelopes, differents Envelopes may needs differents constructo...
DataSetInfo & GetDataLoaderDataSetInfo()
Utility method to get TMVA::DataSetInfo reference from the DataLoader.
void WriteDataInformation(TMVA::DataSetInfo &fDataSetInfo, TMVA::Types::EAnalysisType fAnalysisType)
method to save Train/Test information into the output file.
create variable transformations