From 6aaaaa60c6f0246a11381c3cecacf93bcd9984b1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alexander=20M=C3=A4rz?= Date: Tue, 29 Aug 2023 09:14:55 +0200 Subject: [PATCH] Add support for mixture distributions --- lightgbmlss/distributions/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/lightgbmlss/distributions/__init__.py b/lightgbmlss/distributions/__init__.py index c2bf73a..d1cffc0 100644 --- a/lightgbmlss/distributions/__init__.py +++ b/lightgbmlss/distributions/__init__.py @@ -3,6 +3,7 @@ from . import distribution_utils from . import flow_utils from . import zero_inflated +from . import mixture_distribution_utils from . import Gaussian from . import StudentT from . import Gamma @@ -20,4 +21,5 @@ from . import ZAGamma from . import ZABeta from . import ZALN -from . import SplineFlow \ No newline at end of file +from . import SplineFlow +from . import Mixture \ No newline at end of file