Commit 953ac0ca authored by Baptiste Bauvin's avatar Baptiste Bauvin
Browse files

Added imbalanced bagging

parent 189f6c5a
......@@ -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',
......
......@@ -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",
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment