Update 0.1.3

- Enabled Stacking Ensembler with CV - Removed Comments Signed-off-by: Arsh <[email protected]>
Revca-ANAI · Jul 26, 2022 · 17d4e44 · 17d4e44
1 parent 5eaff73
commit 17d4e44
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 8 deletions.
diff --git a/anai/utils/ensembler.py b/anai/utils/ensembler.py
@@ -36,7 +36,7 @@ def __init__(self, task: str, n_estimators: int = 3, n_clusters: int = 3, estima
         self.est = None
         self.__ensembling_models_reg = {
             'Stacking Ensembler': self._stacking_ensembler,
-            # 'Stacking Ensembler with CV': self._stacking_ensembler_cv,
+            'Stacking Ensembler with CV': self._stacking_ensembler_cv,
             'Max Voting Ensembler': self._max_voting
         }
         self.__ensembling_models_clf = {

diff --git a/anai/utils/explainable_anai/permutation.py b/anai/utils/explainable_anai/permutation.py
@@ -5,12 +5,6 @@
 
 
 def permutational_feature_importance(columns, X_test, y_test, model, isReg=False):
-    # if isinstance(y_test, np.ndarray):
-    #     if len(np.unique(y_test)) > 2 and not isReg:
-    #         raise Exception("Cannot use Permutation Explainers for Multi Class Classification")
-    # else:
-    #     if len(np.unique(y_test.values)) > 2 and not isReg:
-    #         raise Exception("Cannot use Permutation Explainers for Multi Class Classification")
     perm_importance = permutation_importance(
         model,
         X_test,

diff --git a/setup.py b/setup.py
@@ -21,7 +21,7 @@
         ],
         include=["anai.*", "anai"],
     ),
-    version="0.1.2",
+    version="0.1.3",
     license="Apache License 2.0",
     description="Automated ML",
     url="https://github.com/Revca-ANAI/ANAI",