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
36 changes: 23 additions & 13 deletions Graph_Representation_Learning_Rushil_Singha/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,27 +7,29 @@ This project builds **k-nearest neighbor (kNN) jet graphs**, learns **Chebyshev
---

## 🚀 Features

- kNN graph construction from jet particle clouds
- Graph encoder using **Chebyshev GCN** (`SimpleChebNet`)
- Latent **diffusion process** with denoising MLP
- Jet particle **decoder** network
- Evaluation with **KL divergence** & **Wasserstein distance**
- Evaluation using **KL divergence** & **Wasserstein distance**
- Visualization utilities for jet properties

---

## ⚙️ Installation

Clone the repo and install dependencies:
Clone the repository and install dependencies:

```bash

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

pip install -r requirements.txt
```

requirements.txt

```sh
numpy==1.24.3
torch==2.0.0
torch-geometric
Expand All @@ -38,18 +40,26 @@ networkx
scikit-learn
jetnet
```
# This script:

->Encodes jets into latent space
## 📜 Script Overview

->Runs diffusion training
The main script performs the following steps:

->Decodes jets back into particle space

->Logs evaluation metrics

->Saves visualizations to results/
- Encodes jets into latent space
- Runs diffusion training
- Decodes jets back into particle space
- Logs evaluation metrics
- Saves visualizations to results/ directory

## To run the main workflow:
```sh
python code.py
```
## Usage

This repository contains a research prototype for graph-based jet generation using latent diffusion models.

## Note:
Dataset paths, hyperparameters, and output locations may need to be adjusted depending on the local environment and JetNet configuration.