Skip to content

Commit 40e5558

Browse files
authored
DOC fix estimator documentation in IHT (#890)
1 parent e802a19 commit 40e5558

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

imblearn/under_sampling/_prototype_selection/_instance_hardness_threshold.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -38,13 +38,8 @@ class InstanceHardnessThreshold(BaseUnderSampler):
3838
Parameters
3939
----------
4040
estimator : estimator object, default=None
41-
Classifier to be used to estimate instance hardness of the samples. By
42-
default a :class:`~sklearn.ensemble.RandomForestClassifier` will be
43-
used. If ``str``, the choices using a string are the following:
44-
``'knn'``, ``'decision-tree'``, ``'random-forest'``, ``'adaboost'``,
45-
``'gradient-boosting'`` and ``'linear-svm'``. If object, an estimator
46-
inherited from :class:`~sklearn.base.ClassifierMixin` and having an
47-
attribute :func:`predict_proba`.
41+
Classifier to be used to estimate instance hardness of the samples.
42+
This classifier should implement `predict_proba`.
4843
4944
{sampling_strategy}
5045

0 commit comments

Comments
 (0)