bTag-trackSel-optimisation/trainMVA.py at master · BrieucF/bTag-trackSel-optimisation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from tools.MVA_facilities import train_MVA

TrackVars_training = [

    "Track_dz",
    "Track_length",
    "Track_dist",
    "Track_IP2D",
    "Track_pt",
    "Track_chi2",
    "Track_nHitPixel",
    "Track_nHitAll"

]

trackSel = "track_hist_trackFrombJetVStrackNonbJet_no"  #"track_hist_trackFromBVStrackNonbJet_no"  #"track_hist_BfrombVSFake_bTagLoosened" # "bTag_zIP_abs"   #"no" , "bTag", "bTagLoosened", "dist"

absolutePath = "/home/fynu/bfrancois/MyCMSSWcode/bTag/bTag_track_optimisation"
bkgTree = absolutePath + "/bkgTrackTree_"+trackSel+"TrackSel.root"
sigTree = absolutePath + "/sigTrackTree_"+trackSel+"TrackSel.root"


train_MVA(bkgTree, sigTree, TrackVars_training, "BDT", trackSel+"Sel_allVar")