c = Framework()
c.label = "bdt_ucsd_hiStat_cs_ebe"
comment = "Option 3.2: BDTG_UCSD, with even-by-event mass res. weights"
# 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.ebe_weights = True

##################### Input samples #######################

c.add_signal_dir(ggH_2017_powheg.name, ggH_2017_powheg.path,
                 ggH_2017_powheg.xSec)
c.add_signal_dir(VBF_2017_powheg.name, VBF_2017_powheg.path,
                 VBF_2017_powheg.xSec)

c.add_background_dir(ZJets_aMC_2017_hiStat.name, ZJets_aMC_2017_hiStat.path,
                     ZJets_aMC_2017_hiStat.xSec)
c.add_background_dir(tt_ll_POW_2017.name, tt_ll_POW_2017.path,
                     tt_ll_POW_2017.xSec)

##########################################################
c = Framework()
c.label = "bdt_uf_hiStat"
comment = "Option 1: BDTG_UF_v1 (as in HIG-17-019), 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.ebe_weights = False

##################### Input samples #######################

c.add_signal_dir(ggH_2017_powheg.name, ggH_2017_powheg.path, ggH_2017_powheg.xSec)
c.add_signal_dir(VBF_2017_powheg.name, VBF_2017_powheg.path, VBF_2017_powheg.xSec)

c.add_background_dir(ZJets_aMC_2017_hiStat.name, ZJets_aMC_2017_hiStat.path, ZJets_aMC_2017_hiStat.xSec)
c.add_background_dir(tt_ll_POW_2017.name, tt_ll_POW_2017.path, tt_ll_POW_2017.xSec)

##########################################################


###  ------   Raffaele's variables   ------ ###
c.add_variable("muPairs.pt", 				1) 
c.add_variable("muPairs.eta", 				1)