Skip to content

Commit d4b5aa8

Browse files
authored
Updated pymc.DiscreteWeibull docstring (#6283)
* Updated pymc.DiscreteWeibull docstring
1 parent 0be5295 commit d4b5aa8

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

pymc/distributions/discrete.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -462,7 +462,7 @@ def rng_fn(cls, rng, q, beta, size):
462462

463463

464464
class DiscreteWeibull(Discrete):
465-
R"""Discrete Weibull log-likelihood
465+
R"""Discrete Weibull log-likelihood.
466466
467467
The discrete Weibull distribution is a flexible model of count data that
468468
can handle both over- and under-dispersion.
@@ -471,6 +471,7 @@ class DiscreteWeibull(Discrete):
471471
.. math:: f(x \mid q, \beta) = q^{x^{\beta}} - q^{(x + 1)^{\beta}}
472472
473473
.. plot::
474+
:context: close-figs
474475
475476
import matplotlib.pyplot as plt
476477
import numpy as np
@@ -499,6 +500,14 @@ def DiscreteWeibull(q, b, x):
499500
Mean :math:`\mu = \sum_{x = 1}^{\infty} q^{x^{\beta}}`
500501
Variance :math:`2 \sum_{x = 1}^{\infty} x q^{x^{\beta}} - \mu - \mu^2`
501502
======== ======================
503+
504+
Parameters
505+
----------
506+
q : tensor_like of float
507+
Shape parameter (0 < q < 1).
508+
beta : tensor_like of float
509+
Shape parameter (beta > 0).
510+
502511
"""
503512
rv_op = discrete_weibull
504513

0 commit comments

Comments
 (0)