From 17d4e44b5d496f38afdb554a06e0db51eaab403c Mon Sep 17 00:00:00 2001 From: Arsh Date: Tue, 26 Jul 2022 14:37:48 +0530 Subject: [PATCH] Update 0.1.3 - Enabled Stacking Ensembler with CV - Removed Comments Signed-off-by: Arsh --- anai/utils/ensembler.py | 2 +- anai/utils/explainable_anai/permutation.py | 6 ------ setup.py | 2 +- 3 files changed, 2 insertions(+), 8 deletions(-) diff --git a/anai/utils/ensembler.py b/anai/utils/ensembler.py index e5e6864..f6803b8 100644 --- 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 index 9f8a3fb..ee470d8 100644 --- 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 index f7a5d5c..96308b8 100644 --- 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",