Skip to content

Commit

Permalink
[ENH] Half Logistic Distribution (#373)
Browse files Browse the repository at this point in the history
Towards #22, Implements Half Logistic distribution
  • Loading branch information
SaiRevanth25 authored Jun 7, 2024
1 parent 1002f7e commit a40f8f8
Show file tree
Hide file tree
Showing 3 changed files with 83 additions and 0 deletions.
1 change: 1 addition & 0 deletions docs/source/api_reference/distributions.rst
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,7 @@ Continuous support
Fisk
Gamma
HalfCauchy
HalfLogistic
HalfNormal
Laplace
Logistic
Expand Down
2 changes: 2 additions & 0 deletions skpro/distributions/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
"Fisk",
"Gamma",
"HalfCauchy",
"HalfLogistic",
"HalfNormal",
"IID",
"Laplace",
Expand Down Expand Up @@ -42,6 +43,7 @@
from skpro.distributions.fisk import Fisk
from skpro.distributions.gamma import Gamma
from skpro.distributions.halfcauchy import HalfCauchy
from skpro.distributions.halflogistic import HalfLogistic
from skpro.distributions.halfnormal import HalfNormal
from skpro.distributions.laplace import Laplace
from skpro.distributions.logistic import Logistic
Expand Down
80 changes: 80 additions & 0 deletions skpro/distributions/halflogistic.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,80 @@
# copyright: skpro developers, BSD-3-Clause License (see LICENSE file)
"""Half-Logistic probability distribution."""

__author__ = ["SaiRevanth25"]

import pandas as pd
from scipy.stats import halflogistic, rv_continuous

from skpro.distributions.adapters.scipy import _ScipyAdapter


class HalfLogistic(_ScipyAdapter):
r"""Half-Logistic distribution.
This distribution is univariate, without correlation between dimensions
for the array-valued case.
The half-logistic distribution is a continuous probability distribution derived
from the logistic distribution by taking only the positive half. It is particularly
useful in reliability analysis, lifetime modeling, and other applications where
non-negative values are required.
The half-logistic distribution is parametrized by the scale parameter
:math:`\beta`, such that the pdf is
.. math::
f(x) = \frac{2 \exp\left(-\frac{x}{\beta}\right)}
{\beta \left(1 + \exp\left(-\frac{x}{\beta}\right)\right)^2},
x>0 otherwise 0
The scale parameter :math:`\beta` is represented by the parameter ``beta``.
Parameters
----------
beta : float or array of float (1D or 2D), must be positive
scale parameter of the half-logistic distribution
index : pd.Index, optional, default = RangeIndex
columns : pd.Index, optional, default = RangeIndex
Example
-------
>>> from skpro.distributions.halflogistic import HalfLogistic
>>> hl = HalfLogistic(beta=1)
"""

_tags = {
"capabilities:approx": ["pdfnorm"],
"capabilities:exact": ["mean", "var", "pdf", "log_pdf", "cdf", "ppf"],
"distr:measuretype": "continuous",
"distr:paramtype": "parametric",
"broadcast_init": "on",
}

def __init__(self, beta, index=None, columns=None):
self.beta = beta

super().__init__(index=index, columns=columns)

def _get_scipy_object(self) -> rv_continuous:
return halflogistic

def _get_scipy_param(self):
beta = self._bc_params["beta"]
return [beta], {}

@classmethod
def get_test_params(cls, parameter_set="default"):
"""Return testing parameter settings for the estimator."""
# array case examples
params1 = {"beta": [[1, 2], [3, 4]]}
params2 = {
"beta": 1,
"index": pd.Index([1, 2, 5]),
"columns": pd.Index(["a", "b"]),
}
# scalar case examples
params3 = {"beta": 2}
return [params1, params2, params3]

0 comments on commit a40f8f8

Please sign in to comment.