Что в тензорном потоке - PullRequest
       11

Что в тензорном потоке

0 голосов
/ 26 апреля 2020

Я хочу сообщить значение 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.

Спасибо

...