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