Skip to content

Scripts to convert SND@LHC data to pytorch training inputs

Notifications You must be signed in to change notification settings

steggema/snd-scripts

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

78 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Converts SND@LHC digitised simulation files into input data for ML

The conversion has two parts:

  1. SND@LHC digitised ROOT files -> numpy arrays (within SND@LHC experiment software environment)
  2. numpy arrays -> pyg format (pytorch + pyg environment, where pyg is a pytorch GNN library)

The two-step procedure is needed because the SND@LHC SW setup does not contain pytorch/pyg.

SND Digi -> numpy arrays

Main script: digi_to_ml.py runs the conversion based on passed arguments

Condor batch submission: condor contains a condor submission files and scripts to create parameter lists and to merge the outputs

numpy arrays -> pyg format

Main script: convert_to_pyg.py

About

Scripts to convert SND@LHC data to pytorch training inputs

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published