import os, sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from src.classifier import Framework
from samples import *
c = Framework()
c.label = "bdt_dnn_multi_hiStat"
comment = "Option 7: 4-class DNN, w/o even-by-event mass res., ZJets_AMC_hiStat ( mass 105-160 )"
# change this line for each run
c.add_comment(comment)
print comment
c.outDir = '/tmp/dkondrat/ML_output/'
treePath = 'dimuons/tree'
metadataPath = 'dimuons/metadata'
c.set_tree_path(treePath)
c.set_metadata_path(metadataPath)
c.set_year("2017")
c.multiclass = True
c.dy_label = "ZJets_aMC"
c.tt_label = "tt_ll_POW"
c.ggh_label = "H2Mu_gg"
c.vbf_label = "H2Mu_VBF"
##################### Input samples #######################
c.add_data(SingleMu2017B.name, SingleMu2017B.path, SingleMu2017B.lumi)
import os, sys
sys.path.append( os.path.dirname( os.path.dirname( os.path.abspath(__file__) ) ) )
from src.classifier import Framework
c = Framework()
c.add_comment("BDT + singleMu") # change this line for each run!
treePath = 'dimuons/tree'
mc_path = "/mnt/hadoop/store/user/dkondrat/"
signal = [
['H2Mu_VBF', "/VBF_HToMuMu_M125_13TeV_powheg_pythia8/H2Mu_VBF/180827_202716/0000/", 0.0008208 ],
['H2Mu_gg', "/GluGlu_HToMuMu_M125_13TeV_powheg_pythia8/H2Mu_gg/180827_202700/0000/", 0.009618 ],
# ['H2Mu_WH_neg', "/WMinusH_HToMuMu_M125_13TeV_powheg_pythia8/H2Mu_WH_neg/180827_202757/0000/", 0.0001164 ],
# ['H2Mu_WH_pos', "/WPlusH_HToMuMu_M125_13TeV_powheg_pythia8/H2Mu_WH_pos/180827_202738/0000/", 0.0001858 ],
# ['H2Mu_ZH', "/ZH_HToMuMu_M125_13TeV_powheg_pythia8/H2Mu_ZH/180827_202818/0000/", 0.0002136 ]
]
bkg_path = "/mnt/hadoop/store/user/dkondrat/"#"/tmp/Hmumu_ntuples"
background = [
# ['WW', "/WWTo2L2Nu_13TeV-powheg/WW/180827_203218/0000/", 12.46 ],
# ['WWW', "/WWW_4F_TuneCUETP8M1_13TeV-amcatnlo-pythia8/WWW/180827_203402/0000/", 0.2086 ],
# ['WWZ', "/WWZ_TuneCUETP8M1_13TeV-amcatnlo-pythia8/WWZ/180827_203422/0000/", 0.1651 ],
# ['WZZ', "/WZZ_TuneCUETP8M1_13TeV-amcatnlo-pythia8/WZZ/180827_203439/0000/", 0.05565 ],
# ['WZ_2l', "/WZTo2L2Q_13TeV_amcatnloFXFX_madspin_pythia8/WZ_2l/180827_203235/0000/", 4.409 ],
# ['WZ_3l', "/WZTo3LNu_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8/WZ_3l_AMC/180827_203253/0000/", 2.113 ],
# ['ZJets_AMC', "/DYJetsToLL_M-50_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8/ZJets_AMC/180827_202835/0000/", 5765.4 ],
# ['ZZZ', "/ZZZ_TuneCUETP8M1_13TeV-amcatnlo-pythia8/ZZZ/180827_203458/0000/" , 0.01398 ],
# ['ZZ_2l_2q', "/ZZTo2L2Q_13TeV_amcatnloFXFX_madspin_pythia8/ZZ_2l_2q/180827_203327/0000/", 3.22 ],
# ['ZZ_2l_2v', "/ZZTo2L2Nu_13TeV_powheg_pythia8/ZZ_2l_2v/180827_203311/0000/", 0.564 ],
# ['ZZ_4l', "/ZZTo4L_13TeV-amcatnloFXFX-pythia8/ZZ_4l_AMC/180827_203344/0000/", 1.212 ],
