Raynchowkw / L1METML

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

L1METML

Setup

git clone git@github.com:jmduarte/L1METML.git
cd L1METML

#Caution : You need ROOT version over 6.22 to use pyROOT on python 3.

Create an anaconda environment with Python 3.6 and install packages needed to run train.py.

sh conda_setup.sh

Convert

The TTBar sample used in 'convert-uproot.py' is located in

/afs/cern.ch/work/d/daekwon/public/L1PF_110X/CMSSW_11_1_2/src/FastPUPPI/NtupleProducer/python/TTbar_PU200_110X_1M/

or at this CERNBox link: https://cernbox.cern.ch/index.php/s/JK2InUjatHFxFbf

Convert into hdf5

python convertNanoToHDF5_L1triggerToDeepMET.py -i [input .root file path] -o [output file path]

Train

python train.py --dataType ['h5' or 'root': convert root to h5 files or use data generator on root files]  --input [path to input files ending with /] --output [output path (plot and weight will be stored) ending with /] --mode [0 or 1 (0 for L1MET model, 1 for DeepMET model)] --epochs [int]

For example,

python train.py --input ./preprocessed/input.txt --output ./result/ --mode 0

About


Languages

Language:Python 86.8%Language:C 12.6%Language:Dockerfile 0.5%Language:Shell 0.1%