From d2c19d7d30ba2211759faae4a505b45baa456ae3 Mon Sep 17 00:00:00 2001 From: Julia Balla Date: Wed, 12 Jun 2024 19:57:42 -0400 Subject: [PATCH] Update data download instructions --- README.md | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index f2ed411..414a15f 100644 --- a/README.md +++ b/README.md @@ -39,15 +39,18 @@ See [`notebooks/examples.ipynb`](./notebooks/examples.ipynb) for example usage o ## Cosmological benchmark -The TFRecord dataset can be downloaded [here](https://doi.org/10.5281/zenodo.11479419). +The cosmological benchmarking dataset, available in TFRecord format, can be downloaded from Zenodo under the DOI [10.5281/zenodo.11479419](https://doi.org/10.5281/zenodo.11479419). To download the dataset into `benchmarks/galaxies/quijote_records`, run: +```bash +bash benchmarks/galaxies/download_tfrecords.sh +``` To run the graph-level task: ```bash - python benchmarks/galaxies/train_cosmology.py +python benchmarks/galaxies/train_cosmology.py ``` To run the node-level task: ```bash - python benchmarks/galaxies/train_velocities.py +python benchmarks/galaxies/train_velocities.py ``` ## Attribution