h0rton 1.0

Bayesian neural network for hierarchical inference of the Hubble constant

Hierarchical Bayesian inference of the Hubble constant from samples of strong gravitational time delay lenses. This package represents LSST Dark Energy Science Collaboration research in progress. Please contact Ji Won Park (@jiwoncpark) for questions on using the code.

Installation

0. Virtual environments are strongly recommended, to prevent dependencies with conflicting versions. Create a conda virtual environment and activate it:

$conda create -n h0rton python=3.6 -y
$conda activate h0rton

1. You’ll need a Fortran compiler and Fortran-compiled fastell4py, which you can get on a debian system by running

$sudo apt-get install gfortran
$git clone https://github.com/sibirrer/fastell4py.git <desired location>
$cd <desired location>/fastell4py
$python setup.py install --user

2. Now do one of the following.

Option 2(a): clone the repo (please do this if you’d like to contribute to the development).

$git clone https://github.com/jiwoncpark/h0rton.git
$cd h0rton
$pip install -e . -r requirements.txt

Option 2(b): pip install the release version (only recommended if you’re a user).

$pip install h0rton

3. (Optional) To run the notebooks, add the Jupyter kernel.

$python -m ipykernel install --user --name h0rton --display-name "Python (h0rton)"

4. (Optional) To experiment with H0 inference using the TDLMC data, download all their data first using the provided convenience script:

$source h0rton/tdlmc_data/download_tdlmc_data.sh

How to train

1. Generate the training and validation data, e.g.

$python -m baobab.generate h0rton/trainval_data/train_tdlmc_diagonal_config.py

2. Edit the configuration parameters h0rton/example_user_config.py. Make sure the cfg.data field agrees with the training data you generated.

3. Run

$python -m h0rton.train h0rton/example_user_config.py

You can visualize the training results by running

$tensorboard --logdir runs

Feedback and More

Suggestions are always welcome! If you encounter issues or areas for improvement, please message @jiwoncpark or make an issue.

h0rton was used to enter the Time Delay Lens Modeling Challenge:

• Challenge webpage: https://tdlmc.github.io/

• Experimental design paper: https://arxiv.org/abs/1801.01506

This software was developed within the LSST DESC using LSST DESC resources, and so meets the criteria given in, and is bound by, the LSST DESC Publication Policy for being a “DESC product.” When referring to h0rton, please cite (Park et al 2020) and provide a link to this repo.