Python tf.compat.v1.distributions.StudentT用法及代碼示例

學生的 t 分布。

繼承自：Distribution

用法

tf.compat.v1.distributions.StudentT(
    df, loc, scale, validate_args=False, allow_nan_stats=True,
    name='StudentT'
)

此分布具有參數：自由度 df 、位置 loc 和 scale 。

數學細節

概率密度函數 (pdf) 是，

pdf(x; df, mu, sigma) = (1 + y**2 / df)**(-0.5 (df + 1)) / Z
where,
y = (x - mu) / sigma
Z = abs(sigma) sqrt(df pi) Gamma(0.5 df) / Gamma(0.5 (df + 1))

其中:

• loc = mu ,
• scale = sigma ，並且，
• Z 是歸一化常數，並且，
• Gamma 是伽瑪​​函數。

StudentT 分布是location-scale 係列的成員，即它可以構造為，

X ~ StudentT(df, loc=0, scale=1)
Y = loc + scale * X

請注意，scale 的語義更類似於標準差而不是方差。然而，它實際上不是標準。偏差;學生的 t 分布標準。開發。是 scale sqrt(df / (df - 2)) 當 df > 2 。

該分布的樣本被重新參數化(路徑可微)。導數是使用 (Figurnov et al., 2018) 中說明的方法計算的。

例子

一個或一批分布的初始化示例。

import tensorflow_probability as tfp
tfd = tfp.distributions

# Define a single scalar Student t distribution.
single_dist = tfd.StudentT(df=3)

# Evaluate the pdf at 1, returning a scalar Tensor.
single_dist.prob(1.)

# Define a batch of two scalar valued Student t's.
# The first has degrees of freedom 2, mean 1, and scale 11.
# The second 3, 2 and 22.
multi_dist = tfd.StudentT(df=[2, 3], loc=[1, 2.], scale=[11, 22.])

# Evaluate the pdf of the first distribution on 0, and the second on 1.5,
# returning a length two tensor.
multi_dist.prob([0, 1.5])

# Get 3 samples, returning a 3 x 2 tensor.
multi_dist.sample(3)

盡可能廣播參數。

# Define a batch of two Student's t distributions.
# Both have df 2 and mean 1, but different scales.
dist = tfd.StudentT(df=2, loc=1, scale=[11, 22.])

# Evaluate the pdf of both distributions on the same point, 3.0,
# returning a length 2 tensor.
dist.prob(3.0)

計算樣本的梯度 w.r.t.參數：

df = tf.constant(2.0)
loc = tf.constant(2.0)
scale = tf.constant(11.0)
dist = tfd.StudentT(df=df, loc=loc, scale=scale)
samples = dist.sample(5)  # Shape [5]
loss = tf.reduce_mean(tf.square(samples))  # Arbitrary loss function
# Unbiased stochastic gradients of the loss function
grads = tf.gradients(loss, [df, loc, scale])

參考：

隱式重新參數化梯度：Figurnov 等人，2018 (pdf)