diff --git a/README.md b/README.md index 414ea04..2be481b 100644 --- a/README.md +++ b/README.md @@ -48,9 +48,9 @@ LightGBMLSS currently supports the following [PyTorch distributions](https://pyt | [Laplace](https://pytorch.org/docs/stable/distributions.html#laplace) | `Laplace()` | Continuous
(Univariate) | $y \in (-\infty,\infty)$ | 2 | | [LogNormal](https://pytorch.org/docs/stable/distributions.html#lognormal) | `LogNormal()` | Continuous
(Univariate) | $y \in (0,\infty)$ | 2 | | [Negative Binomial](https://pytorch.org/docs/stable/distributions.html#negativebinomial) | `NegativeBinomial()` | Discrete Count
(Univariate) | $y \in (0, 1, 2, 3, \ldots)$ | 2 | -| [Spline Flow](https://docs.pyro.ai/en/stable/distributions.html#pyro.distributions.transforms.Spline) | `SplineFlow()` | Continuous \& Discrete Count
(Univariate) | $y \in (-\infty,\infty)$

$y \in [0, \infty)$

$y \in [0, 1]$

$y \in (0, 1, 2, 3, \ldots)$ | 2xcount_bins + (count_bins-1) (order=quadratic)

3xcount_bins + (count_bins-1) (order=linear) | | [Poisson](https://pytorch.org/docs/stable/distributions.html#poisson) | `Poisson()` | Discrete Count
(Univariate) | $y \in (0, 1, 2, 3, \ldots)$ | 1 | | [Student-T](https://pytorch.org/docs/stable/distributions.html#studentt) | `StudentT()` | Continuous
(Univariate) | $y \in (-\infty,\infty)$ | 3 | +| [Spline Flow](https://docs.pyro.ai/en/stable/distributions.html#pyro.distributions.transforms.Spline) | `SplineFlow()` | Continuous \& Discrete Count
(Univariate) | $y \in (-\infty,\infty)$

$y \in [0, \infty)$

$y \in [0, 1]$

$y \in (0, 1, 2, 3, \ldots)$ | 2xcount_bins + (count_bins-1) (order=quadratic)

3xcount_bins + (count_bins-1) (order=linear) | | [Weibull](https://pytorch.org/docs/stable/distributions.html#weibull) | `Weibull()` | Continuous
(Univariate) | $y \in [0, \infty)$ | 2 | | [Zero-Adjusted Beta](https://github.com/pyro-ppl/pyro/blob/dev/pyro/distributions/zero_inflated.py) | `ZABeta()` | Discrete-Continuous
(Univariate) | $y \in [0, 1)$ | 3 | | [Zero-Adjusted Gamma](https://github.com/pyro-ppl/pyro/blob/dev/pyro/distributions/zero_inflated.py) | `ZAGamma()` | Discrete-Continuous
(Univariate) | $y \in [0, \infty)$ | 3 |