Skip to content

Commit 521de0a

Browse files
author
ThibeauWouters
committed
added utilities files
1 parent e15e705 commit 521de0a

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

src/jimgw/utils/hyperparameters.py

+22
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
jim_default_hyperparameters = {
2+
"seed": 0,
3+
"n_chains": 20,
4+
"num_layers": 10,
5+
"hidden_size": [128,128],
6+
"num_bins": 8,
7+
"local_sampler_arg": {},
8+
"n_walkers_maximize_likelihood": 100,
9+
"n_loops_maximize_likelihood": 200,
10+
}
11+
12+
jim_explanation_hyperparameters = {
13+
"seed": "(int) Value of the random seed used",
14+
"n_chains": "(int) Number of chains to be used",
15+
"num_layers": "(int) Number of hidden layers of the NF",
16+
"hidden_size": "List[int, int] Sizes of the hidden layers of the NF",
17+
"num_bins": "(int) Number of bins used in MaskedCouplingRQSpline",
18+
"local_sampler_arg": "(dict) Additional arguments to be used in the local sampler",
19+
"rng_key_set": "(jnp.array) Key set to be used in PRNG keys",
20+
"n_walkers_maximize_likelihood": "(int) Number of walkers used in the maximization of the likelihood with the evolutionary optimizer",
21+
"n_loops_maximize_likelihood": "(int) Number of loops to run the evolutionary optimizer in the maximization of the likelihood",
22+
}

0 commit comments

Comments
 (0)