doc/v634/tmva102__Testing_8py.html

import ROOT

import pickle


from tmva100_DataPreparation import variables

from tmva101_Training import load_data


# Load data

x, y_true, w = load_data("test_signal.root", "test_background.root")


# Load trained model

File = "tmva101.root"

if (ROOT.gSystem.AccessPathName(File)) :

    ROOT.Info("tmva102_Testing.py", File+"does not exist")

    exit()


bdt = ROOT.TMVA.Experimental.RBDT("myBDT", File)


# Make prediction

y_pred = bdt.Compute(x)


# Compute ROC using sklearn

from sklearn.metrics import roc_curve, auc

false_positive_rate, true_positive_rate, _ = roc_curve(y_true, y_pred, sample_weight=w)

score = auc(false_positive_rate, true_positive_rate)


# Plot ROC

c = ROOT.TCanvas("roc", "", 600, 600)

g = ROOT.TGraph(len(false_positive_rate), false_positive_rate, true_positive_rate)

g.SetTitle("AUC = {:.2f}".format(score))

g.SetLineWidth(3)

g.SetLineColor(ROOT.kRed)

g.Draw("AC")

g.GetXaxis().SetRangeUser(0, 1)

g.GetYaxis().SetRangeUser(0, 1)

g.GetXaxis().SetTitle("False-positive rate")

g.GetYaxis().SetTitle("True-positive rate")

c.Draw()

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

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