Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
PyMethodBase.h
Go to the documentation of this file.
1// @(#)root/tmva/pymva $Id$
2// Authors: Omar Zapata, Lorenzo Moneta, Sergei Gleyzer 2015, Stefan Wunsch 2017
3
4/**********************************************************************************
5 * Project: TMVA - a Root-integrated toolkit for multivariate data analysis *
6 * Package: TMVA *
7 * Class : PyMethodBase *
8 * Web : http://oproject.org *
9 * *
10 * Description: *
11 * Virtual base class for all MVA method based on Python *
12 * *
13 **********************************************************************************/
14
15#ifndef ROOT_TMVA_PyMethodBase
16#define ROOT_TMVA_PyMethodBase
17
18////////////////////////////////////////////////////////////////////////////////
19// //
20// PyMethodBase //
21// //
22// Virtual base class for all TMVA method based on Python //
23// //
24////////////////////////////////////////////////////////////////////////////////
25
26#include "TMVA/MethodBase.h"
27#include "TMVA/Types.h"
28
29#include "Rtypes.h"
30#include "TString.h"
31#include <vector>
32
33class TFile;
34class TGraph;
35class TTree;
36class TDirectory;
37class TSpline;
38class TH1F;
39class TH1D;
40
41#ifndef PyObject_HEAD
42struct _object;
43typedef _object PyObject;
44#define Py_single_input 256
45#endif
46
47namespace TMVA {
48
49 class Ranking;
50 class PDF;
51 class TSpline1;
52 class MethodCuts;
53 class MethodBoost;
54 class DataSetInfo;
55
56 /// Function to find current Python executable
57 /// used by ROOT
58 /// If "Python3" is installed, return "python3"
60
61 class PyMethodBase : public MethodBase {
62
63 friend class Factory;
64 public:
65
66 // default constructur
67 PyMethodBase(const TString &jobName,
68 Types::EMVA methodType,
69 const TString &methodTitle,
70 DataSetInfo &dsi,
71 const TString &theOption = "");
72
73 // constructor used for Testing + Application of the MVA, only (no training),
74 // using given weight file
75 PyMethodBase(Types::EMVA methodType,
76 DataSetInfo &dsi,
77 const TString &weightFile);
78
79 // default destructur
80 virtual ~PyMethodBase();
81 //basic python related function
82 static void PyInitialize();
83 static int PyIsInitialized();
84 static void PyFinalize();
85
86 PyObject *Eval(TString code); // required to parse booking options from string to pyobjects
87 static void Serialize(TString file,PyObject *classifier);
88 static Int_t UnSerialize(TString file,PyObject** obj);
89
90 virtual void Train() = 0;
91 // options treatment
92 virtual void Init() = 0;
93 virtual void DeclareOptions() = 0;
94 virtual void ProcessOptions() = 0;
95 // create ranking
96 virtual const Ranking *CreateRanking() = 0;
97
98 virtual Double_t GetMvaValue(Double_t *errLower = nullptr, Double_t *errUpper = nullptr) = 0;
99
100 Bool_t HasAnalysisType(Types::EAnalysisType type, UInt_t numberClasses, UInt_t numberTargets) = 0;
101 protected:
102 // the actual "weights"
103 virtual void AddWeightsXMLTo(void *parent) const = 0;
104 virtual void ReadWeightsFromXML(void *wghtnode) = 0;
105 virtual void ReadWeightsFromStream(std::istream &) = 0; // backward compatibility
106 virtual void ReadWeightsFromStream(TFile &) {} // backward compatibility
107
108 virtual void ReadModelFromFile() = 0;
109
110 // signal/background classification response for all current set of data
111 virtual std::vector<Double_t> GetMvaValues(Long64_t firstEvt = 0, Long64_t lastEvt = -1, Bool_t logProgress = false) = 0;
112
113 protected:
114 PyObject *fModule; // Module to load
115 PyObject *fClassifier; // Classifier object
116
117 PyObject *fPyReturn; // python return data
118
119 protected:
120 void PyRunString(TString code, TString errorMessage="Failed to run python code", int start=Py_single_input); // runs python code from string in local namespace with error handling
121
122 private:
124 static PyObject *fEval; // eval funtion from python
125 static PyObject *fOpen; // open function for files
126
127 protected:
128 static PyObject *fModulePickle; // Module for model persistence
129 static PyObject *fPickleDumps; // Function to dumps PyObject information into string
130 static PyObject *fPickleLoads; // Function to load PyObject information from string
131
132 static PyObject *fMain; // module __main__ to get namespace local and global
133 static PyObject *fGlobalNS; // global namesapace
134 PyObject *fLocalNS; // local namesapace
135
136 public:
137 static void PyRunString(TString code, PyObject *globalNS, PyObject* localNS); // Overloaded static Python utlity function for running Python code
138 static const char* PyStringAsString(PyObject *string); // Python Utility function for converting a Python String object to const char*
139 static std::vector<size_t> GetDataFromTuple(PyObject *tupleObject); // Function casts Python Tuple object into vector of size_t
140 static std::vector<size_t> GetDataFromList(PyObject *listObject); // Function casts Python List object into vector of size_t
141 static PyObject* GetValueFromDict(PyObject* dict, const char* key); // Function to check for a key in dict and return the associated value if present
142 ClassDef(PyMethodBase, 0) // Virtual base class for all TMVA method
143
144 };
145
146} // namespace TMVA
147
148#endif
_object PyObject
#define Py_single_input
long long Long64_t
Definition RtypesCore.h:80
#define ClassDef(name, id)
Definition Rtypes.h:337
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void char Point_t Rectangle_t WindowAttributes_t Float_t Float_t Float_t Int_t Int_t UInt_t UInt_t Rectangle_t Int_t Int_t Window_t TString Int_t GCValues_t GetPrimarySelectionOwner GetDisplay GetScreen GetColormap GetNativeEvent const char const char dpyName wid window const char font_name cursor keysym reg const char only_if_exist regb h Point_t winding char text const char depth char const char Int_t count const char ColorStruct_t color const char Pixmap_t Pixmap_t PictureAttributes_t attr const char char ret_data h unsigned char height h Atom_t Int_t ULong_t ULong_t unsigned char prop_list Atom_t Atom_t Atom_t Time_t type
Describe directory structure in memory.
Definition TDirectory.h:45
A ROOT file is composed of a header, followed by consecutive data records (TKey instances) with a wel...
Definition TFile.h:53
A TGraph is an object made of two arrays X and Y with npoints each.
Definition TGraph.h:41
1-D histogram with a double per channel (see TH1 documentation)
Definition TH1.h:664
1-D histogram with a float per channel (see TH1 documentation)
Definition TH1.h:621
Class that contains all the data information.
Definition DataSetInfo.h:62
This is the main MVA steering class.
Definition Factory.h:80
Virtual base Class for all MVA method.
Definition MethodBase.h:111
virtual void Train()=0
static std::vector< size_t > GetDataFromTuple(PyObject *tupleObject)
Utility function which retrieves and returns the values of the Tuple object as a vector of size_t.
virtual void AddWeightsXMLTo(void *parent) const =0
static int PyIsInitialized()
Check Python interpreter initialization status.
static std::vector< size_t > GetDataFromList(PyObject *listObject)
Utility function which retrieves and returns the values of the List object as a vector of size_t.
static PyObject * fOpen
virtual void DeclareOptions()=0
virtual void ReadWeightsFromStream(std::istream &)=0
virtual const Ranking * CreateRanking()=0
virtual void ReadModelFromFile()=0
static PyObject * fPickleDumps
virtual void Init()=0
PyObject * Eval(TString code)
Evaluate Python code.
static PyObject * fMain
virtual void ReadWeightsFromXML(void *wghtnode)=0
static void PyInitialize()
Initialize Python interpreter.
static void Serialize(TString file, PyObject *classifier)
Serialize Python object.
static void PyFinalize()
virtual Double_t GetMvaValue(Double_t *errLower=nullptr, Double_t *errUpper=nullptr)=0
static Int_t UnSerialize(TString file, PyObject **obj)
Unserialize Python object.
Bool_t HasAnalysisType(Types::EAnalysisType type, UInt_t numberClasses, UInt_t numberTargets)=0
virtual std::vector< Double_t > GetMvaValues(Long64_t firstEvt=0, Long64_t lastEvt=-1, Bool_t logProgress=false)=0
get all the MVA values for the events of the current Data type
PyObject * fClassifier
static const char * PyStringAsString(PyObject *string)
Returns const char* from Python string in PyObject.
static PyObject * fPickleLoads
virtual void ReadWeightsFromStream(TFile &)
virtual void ProcessOptions()=0
static PyObject * fGlobalNS
static PyObject * fModulePickle
static PyObject * fModuleBuiltin
static PyObject * fEval
static PyObject * GetValueFromDict(PyObject *dict, const char *key)
Utility function which checks if a given key is present in a Python dictionary object and returns the...
void PyRunString(TString code, TString errorMessage="Failed to run python code", int start=256)
Execute Python code from string.
Ranking for variables in method (implementation)
Definition Ranking.h:48
Base class for spline implementation containing the Draw/Paint methods.
Definition TSpline.h:31
Basic string class.
Definition TString.h:139
A TTree represents a columnar dataset.
Definition TTree.h:79
create variable transformations
TString Python_Executable()
Function to find current Python executable used by ROOT If "Python3" is installed,...