Logo ROOT  
Reference Guide
MethodRXGB.h
Go to the documentation of this file.
1// @(#)root/tmva/rmva $Id$
2// Author: Omar Zapata,Lorenzo Moneta, Sergei Gleyzer 2015
3
4/**********************************************************************************
5 * Project: TMVA - a Root-integrated toolkit for multivariate data analysis *
6 * Package: TMVA *
7 * Class : RMethodRXGB *
8 * *
9 * Description: *
10 * R´s Package xgboost method based on ROOTR *
11 * *
12 **********************************************************************************/
13
14#ifndef ROOT_TMVA_RMethodXGB
15#define ROOT_TMVA_RMethodXGB
16
17//////////////////////////////////////////////////////////////////////////
18// //
19// RMethodRXGB //
20// //
21// //
22//////////////////////////////////////////////////////////////////////////
23
24#include "TMVA/RMethodBase.h"
25#include <vector>
26
27namespace TMVA {
28
29 class Factory; // DSMTEST
30 class Reader; // DSMTEST
31 class DataSetManager; // DSMTEST
32 class Types;
33 class MethodRXGB: public RMethodBase {
34
35 public :
36
37 // constructors
38 MethodRXGB(const TString &jobName,
39 const TString &methodTitle,
40 DataSetInfo &theData,
41 const TString &theOption = "");
42
44 const TString &theWeightFile);
45
46
47 ~MethodRXGB(void);
48 void Train();
49 // options treatment
50 void Init();
51 void DeclareOptions();
52 void ProcessOptions();
53 // create ranking
55 {
56 return NULL; // = 0;
57 }
58
59
60 Bool_t HasAnalysisType(Types::EAnalysisType type, UInt_t numberClasses, UInt_t numberTargets);
61
62 // performs classifier testing
63 virtual void TestClassification();
64
65
66 Double_t GetMvaValue(Double_t *errLower = 0, Double_t *errUpper = 0);
67 virtual void MakeClass(const TString &classFileName = TString("")) const; //required for model persistence
69 // the actual "weights"
70 virtual void AddWeightsXMLTo(void * /*parent*/) const {} // = 0;
71 virtual void ReadWeightsFromXML(void * /*wghtnode*/) {} // = 0;
72 virtual void ReadWeightsFromStream(std::istream &) {} //= 0; // backward compatibility
73
74 void ReadModelFromFile();
75
76 // signal/background classification response for all current set of data
77 virtual std::vector<Double_t> GetMvaValues(Long64_t firstEvt = 0, Long64_t lastEvt = -1, Bool_t logProgress = false);
78
79 private :
81 friend class Factory; // DSMTEST
82 friend class Reader; // DSMTEST
83 protected:
84
85
86 //RXGBfunction options
87 //https://github.com/dmlc/xgboost/blob/master/doc/parameter.md
92
93 std::vector<UInt_t> fFactorNumeric; //factors creations
94 //xgboost require a numeric factor then background=0 signal=1 from fFactorTrain
95
104
105
106 // get help message text
107 void GetHelpMessage() const;
108
110 };
111} // namespace TMVA
112#endif
unsigned int UInt_t
Definition: RtypesCore.h:46
bool Bool_t
Definition: RtypesCore.h:63
double Double_t
Definition: RtypesCore.h:59
long long Long64_t
Definition: RtypesCore.h:80
#define ClassDef(name, id)
Definition: Rtypes.h:325
int type
Definition: TGX11.cxx:121
This is a class to pass functions from ROOT to R.
This is a class to get ROOT's objects from R's objects.
Definition: TRObject.h:70
Class that contains all the data information.
Definition: DataSetInfo.h:62
Class that contains all the data information.
This is the main MVA steering class.
Definition: Factory.h:80
virtual void ReadWeightsFromStream(std::istream &)=0
Double_t fEta
Definition: MethodRXGB.h:89
std::vector< UInt_t > fFactorNumeric
Definition: MethodRXGB.h:93
virtual std::vector< Double_t > GetMvaValues(Long64_t firstEvt=0, Long64_t lastEvt=-1, Bool_t logProgress=false)
get all the MVA values for the events of the current Data type
Definition: MethodRXGB.cxx:193
UInt_t fMaxDepth
Definition: MethodRXGB.h:90
virtual void AddWeightsXMLTo(void *) const
Definition: MethodRXGB.h:70
void GetHelpMessage() const
Definition: MethodRXGB.cxx:247
ROOT::R::TRFunctionImport xgbtrain
Definition: MethodRXGB.h:97
MethodRXGB(const TString &jobName, const TString &methodTitle, DataSetInfo &theData, const TString &theOption="")
Definition: MethodRXGB.cxx:50
static Bool_t IsModuleLoaded
Definition: MethodRXGB.h:91
ROOT::R::TRFunctionImport asmatrix
Definition: MethodRXGB.h:102
virtual void TestClassification()
initialization
Definition: MethodRXGB.cxx:166
virtual void ReadWeightsFromStream(std::istream &)
Definition: MethodRXGB.h:72
Bool_t HasAnalysisType(Types::EAnalysisType type, UInt_t numberClasses, UInt_t numberTargets)
Definition: MethodRXGB.cxx:96
virtual void MakeClass(const TString &classFileName=TString("")) const
create reader class for method (classification only at present)
Definition: MethodRXGB.cxx:281
ROOT::R::TRFunctionImport xgbload
Definition: MethodRXGB.h:100
void ReadModelFromFile()
Definition: MethodRXGB.cxx:267
virtual void ReadWeightsFromXML(void *)
Definition: MethodRXGB.h:71
ROOT::R::TRFunctionImport asfactor
Definition: MethodRXGB.h:101
const Ranking * CreateRanking()
Definition: MethodRXGB.h:54
ROOT::R::TRFunctionImport xgbsave
Definition: MethodRXGB.h:99
ROOT::R::TRObject * fModel
Definition: MethodRXGB.h:103
ROOT::R::TRFunctionImport xgbdmatrix
Definition: MethodRXGB.h:98
ROOT::R::TRFunctionImport predict
Definition: MethodRXGB.h:96
DataSetManager * fDataSetManager
Definition: MethodRXGB.h:80
Double_t GetMvaValue(Double_t *errLower=0, Double_t *errUpper=0)
Definition: MethodRXGB.cxx:174
Ranking for variables in method (implementation)
Definition: Ranking.h:48
The Reader class serves to use the MVAs in a specific analysis context.
Definition: Reader.h:64
EAnalysisType
Definition: Types.h:128
Basic string class.
Definition: TString.h:136
create variable transformations