From 953ac0cab9f899fd99457ae3ad6ad29a45b2071d Mon Sep 17 00:00:00 2001 From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr> Date: Fri, 25 Mar 2022 12:27:14 -0400 Subject: [PATCH] Added imbalanced bagging --- setup.py | 3 ++- summit/tests/test_multi_view/test_multiview_utils.py | 2 ++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 2b3ab1a0..edafc1b7 100644 --- a/setup.py +++ b/setup.py @@ -25,7 +25,8 @@ def setup_package(): 'scikit-learn>=0.19.0', 'scipy>=1.3.0', 'six>=1.12.0', 'pandas>=0.23.3', 'pyyaml>=3.12', 'plotly>=4.2.1', 'matplotlib>=3.1.1', 'tabulate>=0.8.6', 'pyscm-ml>=1.0.0', - "randomscm @ git+https://github.com/thibgo/randomscm.git#egg=randomscm"], + "randomscm @ git+https://github.com/thibgo/randomscm.git#egg=randomscm", + "imbalanced-learn"], extras_require={ 'dev': ['pytest', 'pytest-cov'], 'doc': ['sphinx >= 3.0.2', 'numpydoc', 'docutils', 'sphinx-autoapi', diff --git a/summit/tests/test_multi_view/test_multiview_utils.py b/summit/tests/test_multi_view/test_multiview_utils.py index 711b152d..3fca7672 100644 --- a/summit/tests/test_multi_view/test_multiview_utils.py +++ b/summit/tests/test_multi_view/test_multiview_utils.py @@ -76,6 +76,7 @@ class TestFunctions(unittest.TestCase): self.assertEqual(avail, ['adaboost', 'decision_tree', 'gradient_boosting', + 'imbalance_bagging', 'knn', 'lasso', 'random_forest', @@ -90,6 +91,7 @@ class TestFunctions(unittest.TestCase): self.assertEqual(avail, ['adaboost', 'decision_tree', 'gradient_boosting', + 'imbalance_bagging', 'knn', 'random_forest', "random_scm", -- GitLab