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. Changebasedirto location of the HDF5 files andoutput_dirinside the call tosave_modelto specify output location.variables.jl: all relevant variables for trining and plottingplots.jl: plot measures of the network. (change variablebasedirto 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.