Hi,
Nice work with interesting ideas!
If I understand the code correctly. The following code is used as the MH step:
|
rand = jax.random.uniform(sub_key, shape=(p,)) |
|
pos = ((jax.nn.sigmoid(-betas[j] * grad) - rand) > 0).flatten() |
|
new_state = jnp.where(mask, pos, state) |
However, it doesn't change the accuracy performance, if I change it to:
pos = -beta[j]*grad > jnp.zeros((p,))
What do you think about this?
Hi,
Nice work with interesting ideas!
If I understand the code correctly. The following code is used as the MH step:
However, it doesn't change the accuracy performance, if I change it to:
pos = -beta[j]*grad > jnp.zeros((p,))What do you think about this?