Merge pull request #51 from Ab2nour/feat(models)/add-univariate-param…

…etric-gamma feat(models): add gamma model
Ab2nour · Nov 14, 2023 · 9b210bc · 9b210bc
2 parents 9d7408b + 6701e2b
commit 9b210bc
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/modelisation/univariate/parametric/models.py b/src/modelisation/univariate/parametric/models.py
@@ -1,13 +1,21 @@
-from lifelines import ExponentialFitter, LogNormalFitter, WeibullFitter
+from lifelines import (
+    ExponentialFitter,
+    GeneralizedGammaFitter,
+    LogNormalFitter,
+    WeibullFitter,
+)
 
-model_type = LogNormalFitter | ExponentialFitter | WeibullFitter
+model_type = (
+    LogNormalFitter | ExponentialFitter | WeibullFitter | GeneralizedGammaFitter
+)
 
 
 def create_models() -> dict[str, model_type]:
     models = {
         "Weibull": WeibullFitter(),
         "Exponentiel": ExponentialFitter(),
         "Log-normal": LogNormalFitter(),
+        "Gamma-généralisée": GeneralizedGammaFitter(),
     }
 
     return models