Skip to content

Commit a451b47

Browse files
committed
DOC fixes in BalancedBaggingClassifier
1 parent a4947ea commit a451b47

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

imblearn/ensemble/classifier.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ class BalancedBaggingClassifier(BaggingClassifier):
136136
estimators_features_ : list of arrays
137137
The subset of drawn features for each base estimator.
138138
139-
classes_ : array of shape = [n_classes]
139+
classes_ : array, shape (n_classes,)
140140
The classes labels.
141141
142142
n_classes_ : int or list
@@ -145,7 +145,7 @@ class BalancedBaggingClassifier(BaggingClassifier):
145145
oob_score_ : float
146146
Score of the training dataset obtained using an out-of-bag estimate.
147147
148-
oob_decision_function_ : array of shape = [n_samples, n_classes]
148+
oob_decision_function_ : ndarray, shape (n_samples, n_classes)
149149
Decision function computed with out-of-bag estimate on the training
150150
set. If n_estimators is small it might be possible that a data point
151151
was never left out during the bootstrap. In this case,

0 commit comments

Comments
 (0)