From 5fb5364206b05cb4fa191053451abdd31084b9fd Mon Sep 17 00:00:00 2001 From: Guillaume Lemaitre Date: Mon, 17 Jan 2022 21:43:16 +0100 Subject: [PATCH] DOC fix estimator documentation in IHT --- .../_prototype_selection/_instance_hardness_threshold.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/imblearn/under_sampling/_prototype_selection/_instance_hardness_threshold.py b/imblearn/under_sampling/_prototype_selection/_instance_hardness_threshold.py index 30385c861..65aceadb1 100644 --- a/imblearn/under_sampling/_prototype_selection/_instance_hardness_threshold.py +++ b/imblearn/under_sampling/_prototype_selection/_instance_hardness_threshold.py @@ -38,13 +38,8 @@ class InstanceHardnessThreshold(BaseUnderSampler): Parameters ---------- estimator : estimator object, default=None - Classifier to be used to estimate instance hardness of the samples. By - default a :class:`~sklearn.ensemble.RandomForestClassifier` will be - used. If ``str``, the choices using a string are the following: - ``'knn'``, ``'decision-tree'``, ``'random-forest'``, ``'adaboost'``, - ``'gradient-boosting'`` and ``'linear-svm'``. If object, an estimator - inherited from :class:`~sklearn.base.ClassifierMixin` and having an - attribute :func:`predict_proba`. + Classifier to be used to estimate instance hardness of the samples. + This classifier should implement `predict_proba`. {sampling_strategy}