我试图了解如何在 Tensorflow 概率中实现以下模型。
- 角度 ,
theta
在 范围内具有一致的先验概率[-pi / 2, +pi / 2]
; - 方向翻转概率 ,
beta
在 范围内具有一致的先验概率[0, 1]
; theta'
设置为:theta' = theta + pi
有概率beta
; 或者theta' = theta
有概率(1 - beta)
;
- 浓度
c
具有HalfCauchy先验概率;和 - 从von Mises 分布
alpha
中得出一个观察结果,以集中为中心。theta'
c
到目前为止,我尝试过的是
import tensorflow_probability as tfp
import numpy as np
tfd = tfp.distributions
model = tfd.JointDistributionSequential(
[
tfd.Uniform(-np.pi / 2, +np.pi / 2, name='theta'), # theta
tfd.Uniform(0.0, 1.0, name='beta'), # beta
tfd.HalfCauchy(loc=0, scale=1), # c
lambda c, beta, theta: tfd.VonMises(
loc=theta + np.pi * tfd.Binomial(probs=beta),
concentration=c,
name='observed'
), # Observation, alpha
]
)
调用它会在二项式部分出现错误:TypeError: __init__() missing 1 required positional argument: 'total_count'
. 我究竟做错了什么?
2020-03-17 更新
最新代码如下。我仍在尝试找出如何实现模型的第 (3) 部分,即theta
通过添加pi
概率来翻转我的角度方向beta
。对此的任何帮助将不胜感激!到目前为止我所拥有的不起作用,因为我无法将伯努利对象乘以浮点数。
model = tfd.JointDistributionSequential(
[
tfd.Uniform(-np.pi / 2, +np.pi / 2, name='theta'), # theta
tfd.Uniform(0.0, 1.0, name='beta'), # beta
tfd.HalfCauchy(loc=0, scale=1), # c
lambda c, beta, theta: tfd.VonMises(
loc=theta + np.pi * tfd.Bernoulli(probs=beta, dtype=tf.float32),
concentration=c,
name='observed'
), # Observation, alpha
]
)