Skip to content

Commit 7dfa6bd

Browse files
authored
DOC: mention EasyEnsemble in BalancedBaggingClassifier (#448)
1 parent 9fedbcd commit 7dfa6bd

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

imblearn/ensemble/classifier.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,8 @@
2323
sampling_strategy=BaseUnderSampler._sampling_strategy_docstring,
2424
random_state=_random_state_docstring)
2525
class BalancedBaggingClassifier(BaggingClassifier):
26-
"""A Bagging classifier with additional balancing.
26+
"""A Bagging classifier with additional balancing. It is similar to
27+
``EasyEnsemble`` [6]_.
2728
2829
This implementation of Bagging is similar to the scikit-learn
2930
implementation. It includes an additional step to balance the training set
@@ -146,6 +147,10 @@ class BalancedBaggingClassifier(BaggingClassifier):
146147
.. [5] Chen, Chao, Andy Liaw, and Leo Breiman. "Using random forest to
147148
learn imbalanced data." University of California, Berkeley 110,
148149
2004.
150+
.. [6] X. Y. Liu, J. Wu and Z. H. Zhou, "Exploratory Undersampling for
151+
Class-Imbalance Learning," in IEEE Transactions on Systems, Man, and
152+
Cybernetics, Part B (Cybernetics), vol. 39, no. 2, pp. 539-550,
153+
April 2009.
149154
150155
Examples
151156
--------

0 commit comments

Comments
 (0)