Add support for mixture distributions

StatMixedML · Aug 29, 2023 · 6aaaaa6 · 6aaaaa6
1 parent b3e83a9
commit 6aaaaa6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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
+from . import SplineFlow
+from . import Mixture