Skip to content
Snippets Groups Projects
README.md 775 B
Newer Older
Bryan Bliewert's avatar
Bryan Bliewert committed
# GraphJet

## Description
Bryan Bliewert's avatar
Bryan Bliewert committed
Goal: Clustering of PandoraPFOs into jets based on TrueJet data
Bryan Bliewert's avatar
Bryan Bliewert committed

## Installation

nVentis's avatar
nVentis committed
### PyTorch (PyTorch Geometric, PyG) environment

`mamba create -n graphjet_pyg python=3.11 -c conda-forge && mamba activate graphjet_pyg && mamba install jupyterlab tqdm ipywidgets==7.7.2 pip matplotlib numpy seaborn uproot tabulate && pip install pyvis pyg torch`
nVentis's avatar
nVentis committed

For documentation on PyTorch Geometric/PyG, see [here](https://pytorch-geometric.readthedocs.io/en/latest/).

## Dataset
nVentis's avatar
nVentis committed

Fully connected graph of final state particles with nodes consisting of four momenta and edges as [0,1] labels displaying the connection of the elements. Objective: Edge prediction.
Bryan Bliewert's avatar
Bryan Bliewert committed

Bryan Bliewert's avatar
Bryan Bliewert committed
## Usage
Bryan Bliewert's avatar
Bryan Bliewert committed

Bryan Bliewert's avatar
Bryan Bliewert committed
## Roadmap
nVentis's avatar
nVentis committed

- Incorporate Particle ID?
Bryan Bliewert's avatar
Bryan Bliewert committed

## Authors and acknowledgment
Bryan Bliewert's avatar
Bryan Bliewert committed
Bryan Bliewert