Python Experiment示例

示例#1

文件： test_event_sample_dump.py 项目： thaarres/VAnDe

import os
import POfAH.util.experiment as ex
import util.event_sample as es
import analysis.analysis_constituents as ac
from config import *

experiment = ex.Experiment(run_n=0)

# read in data
test_sample = es.EventSample.from_input_file(
    'RS Graviton WW br 3.0TeV',
    '/home/kinga/mnt/data/events/RSGraviton_WW_NARROW_13TeV_PU40_3.0TeV_concat_10K.h5'
)
test_evts_j1, test_evts_j2 = test_sample.get_particles()

# setup analysis
particle_analysis = ac.AnalysisConstituents(
    'RS Graviton WW br 3.0TeV',
    fig_dir='/home/kinga/mnt/fig/run_0/analysis_event')
particle_analysis.analyze([test_evts_j1, test_evts_j2])

# dump sample
out_path = os.path.join('/home/kinga/mnt/results/run_0')
test_sample.dump(out_path)

# read back
test_sample = es.EventSample.from_input_file(
    'RS Graviton WW br 3.0TeV dumped',
    os.path.join(out_path, test_sample.file_name))
test_evts_j1, test_evts_j2 = test_sample.get_particles()

示例#2

文件： main_3d_vae.py 项目： thaarres/VAnDe

import os
from config import *
#import setGPU
import numpy as np

import POfAH.util.experiment as ex
import POfAH.util.input_data_reader as idr
import analysis.analysis_constituents as ac
from vae.vae_3Dloss_model import VAE_3D

# ********************************************************
#       runtime params
# ********************************************************

run_n = 4
experiment = ex.Experiment(run_n).setup(model_dir=True, fig_dir=True)

# ********************************************************
#       read in training data ( events )
# ********************************************************

data_reader = idr.InputDataReader(
    os.path.join(config['input_dir'], 'background_small.h5'))
train_evts_j1, train_evts_j2 = data_reader.read_jet_constituents()

particle_analysis = ac.AnalysisConstituents('QCD train',
                                            fig_dir=experiment.fig_dir_event)
particle_analysis.analyze([train_evts_j1, train_evts_j2])

# ********************************************************
#       prepare training data

示例#3

文件： main_analysis_jet_features.py 项目： thaarres/VAnDe

import os

import POfAH.sample_dict as sd
import analysis.analysis_jet_feature as ajf
import POfAH.util.experiment as ex
import POfAH.jet_sample as js
import util.event_sample as es
import util.plotting as up
import POfAH.util.sample_factory as sf

run_n = 101
data_sample = 'particle-local'

experiment = ex.Experiment(run_n).setup(fig_dir=True)
paths = sf.SamplePathFactory(experiment, data_sample)

# ********************************************************
#       read in training data ( events )
# ********************************************************

qcd_sample = js.JetSample.from_input_file('qcdSide', paths.qcd_file_path)
scikit_qcd_sample = ajf.dijet_sample_from_dijet_sample(qcd_sample)
ptjj = [j.pt for j in scikit_qcd_sample]

up.plot_hist(ptjj,
             title='ptjj',
             plot_name='ptjj_dist',
             fig_dir=experiment.fig_dir,
             xlim=(-1, 10),
             bins=3000)