Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
41 changes: 19 additions & 22 deletions Graph_Representation_Learning_Rushil_Singha/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

A PyTorch/PyTorch-Geometric implementation of a **graph-based diffusion model** for generating realistic jets from the [JetNet dataset](https://huggingface.co/datasets/jetnet).

This project builds **k-nearest neighbor (kNN) jet graphs**, learns **Chebyshev GCN (ChebNet) embeddings**, trains a **diffusion model in latent space**, and decodes back into particle-level jets.
This model builds **k-nearest neighbor (kNN) jet graphs**, learns **Chebyshev GCN (ChebNet) embeddings**, trains a **diffusion model in latent space**, and decodes generated samples back into particle-level jets.

---

Expand All @@ -18,37 +18,34 @@ This project builds **k-nearest neighbor (kNN) jet graphs**, learns **Chebyshev

## ⚙️ Installation

Clone the repo and install dependencies:
Clone the ML4Sci GENIE repository and navigate to this project directory :

```bash
git clone https://github.com/your-username/jetnet-graph-diffusion.git
cd jetnet-graph-diffusion

git clone https://github.com/ML4SCI/GENIE.git
cd GENIE/Graph_Representation_Learning_Rushil_Singha
```
Install dependencies:
```bash
pip install -r requirements.txt
```
## 🏃 Quick Start

After installing dependencies, run:

requirements.txt

numpy==1.24.3
torch==2.0.0
torch-geometric
torch-scatter
torch-sparse
torch-cluster
networkx
scikit-learn
jetnet
```bash
python code.py
```
# This script:
# This script will:

->Encodes jets into latent space
- Encodes jets into latent space

->Runs diffusion training
- Runs diffusion training

->Decodes jets back into particle space
- Decodes jets back into particle space

->Logs evaluation metrics
- Logs evaluation metrics

->Saves visualizations to results/
- Saves visualizations to results/



Expand Down