Skip to content

wz1070/jets

 
 

Repository files navigation

Graph neural nets for jet physics

Instructions

Requirements

  • python 3
  • pytorch
  • scikit-learn

Email results

Change the RECIPIENT default in constants.py to your email address. Then you will get emails with results and logfiles. Otherwise you won't.

Data

You need to unzip the tars and put the raw pickle files into data/w-vs-qcd/pickles/raw (make this directory). The training script will look for data in data/w-vs-qcd/pickles/preprocessed and if it doesn't find it, make it from the raw stuff.

Usage

Classification of W vs QCD jets:

# Training
python train.py [argparse args]
# Test
python evaluation.py [argparse args]

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 93.7%
  • Python 6.1%
  • Other 0.2%