Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ENH] Half Logistic Distribution #373

Merged
merged 2 commits into from
Jun 7, 2024
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
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 @@ -41,6 +42,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
fkiraly marked this conversation as resolved.
Show resolved Hide resolved
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]
Loading