From 521de0a64c82d98ed78271b9b5d04273dc124d01 Mon Sep 17 00:00:00 2001 From: ThibeauWouters Date: Sat, 3 Feb 2024 04:27:04 -0800 Subject: [PATCH] added utilities files --- src/jimgw/utils/hyperparameters.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 src/jimgw/utils/hyperparameters.py diff --git a/src/jimgw/utils/hyperparameters.py b/src/jimgw/utils/hyperparameters.py new file mode 100644 index 00000000..ff6f01df --- /dev/null +++ b/src/jimgw/utils/hyperparameters.py @@ -0,0 +1,22 @@ +jim_default_hyperparameters = { + "seed": 0, + "n_chains": 20, + "num_layers": 10, + "hidden_size": [128,128], + "num_bins": 8, + "local_sampler_arg": {}, + "n_walkers_maximize_likelihood": 100, + "n_loops_maximize_likelihood": 200, +} + +jim_explanation_hyperparameters = { + "seed": "(int) Value of the random seed used", + "n_chains": "(int) Number of chains to be used", + "num_layers": "(int) Number of hidden layers of the NF", + "hidden_size": "List[int, int] Sizes of the hidden layers of the NF", + "num_bins": "(int) Number of bins used in MaskedCouplingRQSpline", + "local_sampler_arg": "(dict) Additional arguments to be used in the local sampler", + "rng_key_set": "(jnp.array) Key set to be used in PRNG keys", + "n_walkers_maximize_likelihood": "(int) Number of walkers used in the maximization of the likelihood with the evolutionary optimizer", + "n_loops_maximize_likelihood": "(int) Number of loops to run the evolutionary optimizer in the maximization of the likelihood", +}