Skip to content
Snippets Groups Projects
Commit 953ac0ca authored by Baptiste Bauvin's avatar Baptiste Bauvin
Browse files

Added imbalanced bagging

parent 189f6c5a
Branches
No related tags found
No related merge requests found
...@@ -25,7 +25,8 @@ def setup_package(): ...@@ -25,7 +25,8 @@ def setup_package():
'scikit-learn>=0.19.0', 'scipy>=1.3.0', 'six>=1.12.0', 'scikit-learn>=0.19.0', 'scipy>=1.3.0', 'six>=1.12.0',
'pandas>=0.23.3', 'pyyaml>=3.12', 'plotly>=4.2.1', 'pandas>=0.23.3', 'pyyaml>=3.12', 'plotly>=4.2.1',
'matplotlib>=3.1.1', 'tabulate>=0.8.6', 'pyscm-ml>=1.0.0', '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={ extras_require={
'dev': ['pytest', 'pytest-cov'], 'dev': ['pytest', 'pytest-cov'],
'doc': ['sphinx >= 3.0.2', 'numpydoc', 'docutils', 'sphinx-autoapi', 'doc': ['sphinx >= 3.0.2', 'numpydoc', 'docutils', 'sphinx-autoapi',
......
...@@ -76,6 +76,7 @@ class TestFunctions(unittest.TestCase): ...@@ -76,6 +76,7 @@ class TestFunctions(unittest.TestCase):
self.assertEqual(avail, ['adaboost', self.assertEqual(avail, ['adaboost',
'decision_tree', 'decision_tree',
'gradient_boosting', 'gradient_boosting',
'imbalance_bagging',
'knn', 'knn',
'lasso', 'lasso',
'random_forest', 'random_forest',
...@@ -90,6 +91,7 @@ class TestFunctions(unittest.TestCase): ...@@ -90,6 +91,7 @@ class TestFunctions(unittest.TestCase):
self.assertEqual(avail, ['adaboost', self.assertEqual(avail, ['adaboost',
'decision_tree', 'decision_tree',
'gradient_boosting', 'gradient_boosting',
'imbalance_bagging',
'knn', 'knn',
'random_forest', 'random_forest',
"random_scm", "random_scm",
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment