Mangrove: Infer galaxy properties using dark matter merger trees
Abstract
Mangrove uses Graph Neural Networks to regress baryonic properties directly from full dark matter merger trees to infer galaxy properties. The package includes code for preprocessing the merger tree, and training the model can be done either as single experiments or as a sweep. Mangrove provides loss functions, learning rate schedulers, models, and a script for doing the training on a GPU.
- Publication:
-
Astrophysics Source Code Library
- Pub Date:
- June 2023
- Bibcode:
- 2023ascl.soft06015J
- Keywords:
-
- Software