Julia implementation of Lorentz layer networks inspired by https://arxiv.org/abs/1707.08966
Start Julia with julia --project=script
to automatically get correct versions of all dependencies required.
run.jl
: main entry point for network training. Changebasedir
to location of the HDF5 files andoutput_dir
inside the call tosave_model
to specify output location.variables.jl
: all relevant variables for trining and plottingplots.jl
: plot measures of the network. (change variablebasedir
to location of model output)likelihood.jl
: perform likelihood analyses and plot results.plot_input_dists.jl
/plot_recorded_measures.jl
: plot input distributions/recorded measures
Package should be loaded automatically if specifying script
as the project. Can be imported with using JuliaForHEP
.
hdf_preprocessing.py
: quick and dirty Python script using Pandas + PyROOT, which extracts Cartesian four momenta from detector coordinates and filters out events with != 6 jets.