From c77faaca0014f39bd691b3fe956a6de4fa513e98 Mon Sep 17 00:00:00 2001
From: luigifvr <40857315+luigifvr@users.noreply.github.com>
Date: Wed, 6 Dec 2023 17:43:54 +0100
Subject: [PATCH] Update README.md

---
 README.md | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/README.md b/README.md
index b6d3573..8ab837d 100644
--- a/README.md
+++ b/README.md
@@ -4,6 +4,19 @@ This tool makes it easy to train a classifier to reweight samples from a generat
 model. It comes with an extensive plotting pipeline that allows to inspect the classifier
 output to evaluate the performance of the generative network.
 
+This is the reference repository for the article "How to Understand Limitations of Generative Networks"
+The preprint is available at [https://arxiv.org/abs/2305.16774](https://arxiv.org/abs/2305.16774)
+
+We collect here all the datasets used to train the classifiers.
+Samples obtained from the generative models and truth generated events:
+[![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.10277550.svg)](https://doi.org/10.5281/zenodo.10277550)
+
+Original JetNet dataset:
+Kansal, R., Duarte, J., Su, H., Orzari, B., Tomei, T., Pierini, M., Touranakou, M., Vlimant, J.-R., & Gunopulos, D. (2022). JetNet (Versione 2). Zenodo. [![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.6975118.svg)](https://doi.org/10.5281/zenodo.6975118)
+
+CaloGAN dataset:
+Krause, C., & Shih, D. (2021). Electromagnetic Calorimeter Shower Images of CaloFlow. Zenodo. [![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.5904188.svg)](https://doi.org/10.5281/zenodo.5904188)
+
 ## Usage
 
 Training a discriminator: