fix ENet regressor

Techtonique · Apr 27, 2024 · 51dcff6 · 51dcff6
1 parent 4da5619
commit 51dcff6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/mlsauce/__init__.py b/mlsauce/__init__.py
@@ -23,7 +23,7 @@
 # Dev branch marker is: 'X.Y.dev' or 'X.Y.devN' where N is an integer.
 # 'X.Y.dev0' is the canonical version of 'X.Y.dev'
 #
-__version__ = "0.10.0"
+#__version__ = "0.10.0"
 
 
 # On OSX, we can get a runtime error due to multiple OpenMP libraries loaded
@@ -58,6 +58,7 @@
     from .adaopt import AdaOpt
     from .booster import LSBoostClassifier, LSBoostRegressor
     from .datasets import download
+    from .elasticnet import ElasticNetRegressor
     from .lasso import LassoRegressor
     from .ridge import RidgeRegressor
     from .stump import StumpClassifier
@@ -68,6 +69,7 @@
         "AdaOpt",
         "LSBoostClassifier",
         "StumpClassifier",
+        "ElasticNetRegressor",
         "LassoRegressor",
         "LSBoostRegressor",
         "RidgeRegressor",

diff --git a/mlsauce/booster/_boosterc.pyx b/mlsauce/booster/_boosterc.pyx
@@ -306,7 +306,7 @@ def fit_booster_regressor(double[:,::1] X, double[:] y,
   elif solver == "lasso": 
     fit_obj = LassoRegressor(reg_lambda = reg_lambda, backend = backend)
   else: 
-    fit_obj = LassoRegressor(reg_lambda = reg_lambda, alpha = alpha, backend = backend)  
+    fit_obj = ElasticNetRegressor(reg_lambda = reg_lambda, alpha = alpha, backend = backend)  
 
 
   for iter in iterator: