Python Framework示例

示例#1

import os, sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import argparse
from classifier import Framework

from samples.ntuples import *

parser = argparse.ArgumentParser(description='')
parser.add_argument('--out_path',
                    action='store',
                    dest='output_path',
                    help='Output path')
args = parser.parse_args()

c = Framework(outPath=args.output_path)
c.label = "hammer"
comment = "hammer test"
# change this line for each run
c.add_comment(comment)
print comment

treePath = 'tree'

c.set_tree_path(treePath)

c.set_year("ucsd_inclusive")
c.massWindow = [120, 130]
c.multiclass = True
c.dy_label = "DY"
c.top_label = "ttbar"
c.ggh_label = "ggH"

示例#2

文件： test_latinos.py 项目： kondratyevd/H2MuPurdue

import os, sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import argparse
from classifier import Framework

from samples.ntuples import *

parser = argparse.ArgumentParser(description='')
parser.add_argument('--out_path',
                    action='store',
                    dest='output_path',
                    help='Output path')
args = parser.parse_args()

c = Framework(outPath=args.output_path)
c.label = "latinos"
comment = "latinos test"
# change this line for each run
c.add_comment(comment)
print comment

treePath = 'Events'

c.set_tree_path(treePath)

c.set_year("latinos")
c.massWindow = [120, 130]
c.multiclass = True
c.dy_label = "DY"
c.top_label = "ttbar"
c.ggh_label = "ggH"

示例#3

文件： dnn_binary.py 项目： kondratyevd/H2MuPurdue

import os, sys
sys.path.append( os.path.dirname( os.path.dirname( os.path.abspath(__file__) ) ) )
import argparse
from classifier import Framework

from samples.ntuples import *

parser = argparse.ArgumentParser(description='')
parser.add_argument('--out_path', action='store', dest='output_path', help='Output path')
args = parser.parse_args()


c = Framework(outPath=args.output_path)
c.label = "dnn_resweights"
comment = "DNN UCSD resweights"  
                # change this line for each run
c.add_comment(comment)
print comment

treePath = 'tree'

c.set_tree_path(treePath)

c.set_year("ucsd_inclusive")
c.massWindow = [120,130]
c.multiclass = False


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

示例#4

文件： apply_caltech.py 项目： kondratyevd/H2MuPurdue

import os, sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import argparse
from classifier import Framework

from samples.ntuples import *

parser = argparse.ArgumentParser(description='')
parser.add_argument('--out_path',
                    action='store',
                    dest='output_path',
                    help='Output path')
args = parser.parse_args()

c = Framework(outPath=args.output_path)
c.label = "apply_caltech"
comment = "apply_caltech"
# change this line for each run
c.add_comment(comment)
print comment

treePath = 'Events'

c.set_tree_path(treePath)

c.set_year("caltech")
# c.massWindow = [120,130]

fwk_path = os.environ['FRAMEWORK_PATH']
c.trained_model_path = fwk_path + '/mva/data/DNN27vars_sig_vbf_bkg_dyvbf_dy105To160_ewk105To160_split_60_40_mod10_190820.h5'
c.standartization_path = fwk_path + '/mva/data/DNN27vars_sig_vbf_bkg_dyvbf_dy105To160_ewk105To160_split_60_40_mod10_190820.npy'