speed up (only for big use-cases with many features

ing-bank · Mar 28, 2024 · 0fdc746 · 0fdc746
1 parent 22b4bc5
commit 0fdc746
Show file tree

Hide file tree

Showing 3 changed files with 889 additions and 881 deletions.
diff --git a/probatus/feature_elimination/__init__.py b/probatus/feature_elimination/__init__.py
@@ -1,3 +1,4 @@
-from .feature_elimination import ShapRFECV, EarlyStoppingShapRFECV
+from .feature_elimination import ShapRFECV
+from .early_stopping_feature_elimination import EarlyStoppingShapRFECV
 
 __all__ = ["ShapRFECV", "EarlyStoppingShapRFECV"]