Skip to content

Commit 0a93fc9

Browse files
authored
DOC Clarify what can be passed to the plotting utilities (scikit-learn#16893)
1 parent f31e25b commit 0a93fc9

File tree

3 files changed

+6
-3
lines changed

3 files changed

+6
-3
lines changed

sklearn/metrics/_plot/confusion_matrix.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,8 @@ def plot_confusion_matrix(estimator, X, y_true, labels=None,
137137
Parameters
138138
----------
139139
estimator : estimator instance
140-
Trained classifier.
140+
Fitted classifier or a fitted :class:`~sklearn.pipeline.Pipeline`
141+
in which the last estimator is a classifier.
141142
142143
X : {array-like, sparse matrix} of shape (n_samples, n_features)
143144
Input values.

sklearn/metrics/_plot/precision_recall_curve.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,8 @@ def plot_precision_recall_curve(estimator, X, y,
106106
Parameters
107107
----------
108108
estimator : estimator instance
109-
Trained classifier.
109+
Fitted classifier or a fitted :class:`~sklearn.pipeline.Pipeline`
110+
in which the last estimator is a classifier.
110111
111112
X : {array-like, sparse matrix} of shape (n_samples, n_features)
112113
Input values.

sklearn/metrics/_plot/roc_curve.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,8 @@ def plot_roc_curve(estimator, X, y, sample_weight=None,
115115
Parameters
116116
----------
117117
estimator : estimator instance
118-
Trained classifier.
118+
Fitted classifier or a fitted :class:`~sklearn.pipeline.Pipeline`
119+
in which the last estimator is a classifier.
119120
120121
X : {array-like, sparse matrix} of shape (n_samples, n_features)
121122
Input values.

0 commit comments

Comments
 (0)