From f946696f36ffe2e287aad128a287005a41d5de0d Mon Sep 17 00:00:00 2001 From: jpaillard Date: Wed, 16 Oct 2024 14:45:22 +0200 Subject: [PATCH] remove `TransformerMixin` inheritance --- hidimstat/cpi.py | 4 ++-- hidimstat/loco.py | 4 ++-- hidimstat/permutation_importance.py | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/hidimstat/cpi.py b/hidimstat/cpi.py index 8835fb8..815652f 100644 --- a/hidimstat/cpi.py +++ b/hidimstat/cpi.py @@ -1,10 +1,10 @@ import numpy as np from joblib import Parallel, delayed -from sklearn.base import BaseEstimator, TransformerMixin, check_is_fitted, clone +from sklearn.base import BaseEstimator, check_is_fitted, clone from sklearn.metrics import root_mean_squared_error -class CPI(BaseEstimator, TransformerMixin): +class CPI(BaseEstimator): """ Conditional Permutation Importance (CPI) algorithm. :footcite:t:`Chamma_NeurIPS2023` and for group-level see diff --git a/hidimstat/loco.py b/hidimstat/loco.py index e991469..6d691a3 100644 --- a/hidimstat/loco.py +++ b/hidimstat/loco.py @@ -1,10 +1,10 @@ import numpy as np from joblib import Parallel, delayed -from sklearn.base import BaseEstimator, TransformerMixin, check_is_fitted, clone +from sklearn.base import BaseEstimator, check_is_fitted, clone from sklearn.metrics import root_mean_squared_error -class LOCO(BaseEstimator, TransformerMixin): +class LOCO(BaseEstimator): """ Leave-One-Covariate-Out (LOCO) algorithm as described in :footcite:t:`Chamma_NeurIPS2023`. diff --git a/hidimstat/permutation_importance.py b/hidimstat/permutation_importance.py index 6010f71..f6fc496 100644 --- a/hidimstat/permutation_importance.py +++ b/hidimstat/permutation_importance.py @@ -1,10 +1,10 @@ import numpy as np from joblib import Parallel, delayed -from sklearn.base import BaseEstimator, TransformerMixin, check_is_fitted +from sklearn.base import BaseEstimator, check_is_fitted from sklearn.metrics import root_mean_squared_error -class PermutationImportance(BaseEstimator, TransformerMixin): +class PermutationImportance(BaseEstimator): """