Я хочу сообщить значение rhat для каждого параметра в модели. Предположим, у меня есть 2 параметра, сгруппированных в двухэлементном векторе, скажем,
param_ = np.array([1.,1.], np.float32)
initial_chain_state = tf.convert_to_tensor(param_, dtype=tf.float32)
sample_chains = tfp.mcmc.sample_chain(num_results=np.int(num_results), num_burnin_steps=np.int(num_burnin_steps), parallel_iterations=2, current_state=initial_chain_state, kernel=adaptive_hmc, trace_fn=None)
, который используется для инициализации цепочек, как показано в коде выше.
Я вычисляю значение rhat, используя код:
rhat = tfp.mcmc.diagnostic.potential_scale_reduction(chains_states=sample_chains,
independent_chain_ndims=1,
split_chains=False)
Хотя оценки параметров верны, слишком высокое значение порядка 1e6. Может кто-нибудь, пожалуйста, помогите. Может быть, я делаю это не так, но я уже пытался установить independent_chain_ndims и split_chains.
Спасибо