doc/v630/ApplicationRegressionKeras_8py.html

from ROOT import TMVA, TFile, TString

from array import array

from subprocess import call

from os.path import isfile


# Setup TMVA

TMVA.Tools.Instance()

TMVA.PyMethodBase.PyInitialize()

reader = TMVA.Reader("Color:!Silent")


# Load data

if not isfile('tmva_reg_example.root'):

    call(['curl', '-L', '-O', 'http://root.cern.ch/files/tmva_reg_example.root'])


data = TFile.Open('tmva_reg_example.root')

tree = data.Get('TreeR')


branches = {}

for branch in tree.GetListOfBranches():

    branchName = branch.GetName()

    branches[branchName] = array('f', [-999])

    tree.SetBranchAddress(branchName, branches[branchName])

    if branchName != 'fvalue':

        reader.AddVariable(branchName, branches[branchName])


# Book methods

reader.BookMVA('PyKeras', TString('dataset/weights/TMVARegression_PyKeras.weights.xml'))


# Print some example regressions

print('Some example regressions:')

for i in range(20):

    tree.GetEntry(i)

    print('True/MVA value: {}/{}'.format(branches['fvalue'][0],reader.EvaluateMVA('PyKeras')))

format
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 format
Definition TGWin32VirtualXProxy.cxx:249

TFile::Open
static TFile * Open(const char *name, Option_t *option="", const char *ftitle="", Int_t compress=ROOT::RCompressionSetting::EDefaults::kUseCompiledDefault, Int_t netopt=0)
Create / open a file.
Definition TFile.cxx:4075

TMVA::PyMethodBase::PyInitialize
static void PyInitialize()
Initialize Python interpreter.
Definition PyMethodBase.cxx:153

TMVA::Reader
The Reader class serves to use the MVAs in a specific analysis context.
Definition Reader.h:64

TMVA::Tools::Instance
static Tools & Instance()
Definition Tools.cxx:71

TString
Basic string class.
Definition TString.h:139