From 6ec62c1cd1dab6685a5eb6d028badb26c82ed5fb Mon Sep 17 00:00:00 2001 From: veerlosar Date: Sat, 12 Oct 2019 13:17:07 +0200 Subject: [PATCH 1/3] one_vs_rest_example --- sklearn/multiclass.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/sklearn/multiclass.py b/sklearn/multiclass.py index c09e34b1e9504..00c5c07b9032d 100644 --- a/sklearn/multiclass.py +++ b/sklearn/multiclass.py @@ -181,6 +181,22 @@ class OneVsRestClassifier(MultiOutputMixin, ClassifierMixin, multilabel_ : boolean Whether a OneVsRestClassifier is a multilabel classifier. + >>> import numpy as np + >>> from sklearn.multiclass import OneVsRestClassifier + >>> from sklearn.svm import SVC + >>> X = np.array([ + ... [10, 10], + ... [8, 10], + ... [-5, 5.5], + ... [-5.4, 5.5], + ... [-20, -20], + ... [-15, -20] + ... ]) + >>> y = np.array([0, 0, 1, 1, 2, 2]) + >>> clf = OneVsRestClassifier(SVC()).fit(X, y) + >>> clf.predict([[-19, -20], [-5, 5], [9, 9]]) + array([2, 1, 0]) + """ def __init__(self, estimator, n_jobs=None): self.estimator = estimator From 14da649660f8bf2cdabe979bbbf180b26166fde4 Mon Sep 17 00:00:00 2001 From: veerlosar <45338519+veerlosar@users.noreply.github.com> Date: Sun, 13 Oct 2019 10:54:22 +0200 Subject: [PATCH 2/3] Update sklearn/multiclass.py Co-Authored-By: Thomas J Fan --- sklearn/multiclass.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sklearn/multiclass.py b/sklearn/multiclass.py index 00c5c07b9032d..45dfc1df60e2f 100644 --- a/sklearn/multiclass.py +++ b/sklearn/multiclass.py @@ -194,7 +194,7 @@ class OneVsRestClassifier(MultiOutputMixin, ClassifierMixin, ... ]) >>> y = np.array([0, 0, 1, 1, 2, 2]) >>> clf = OneVsRestClassifier(SVC()).fit(X, y) - >>> clf.predict([[-19, -20], [-5, 5], [9, 9]]) + >>> clf.predict([[-19, -20], [9, 9], [-5, 5]]) array([2, 1, 0]) """ From 35a6c3ea5fea2b61346f57ac54a7ae8dca389bdc Mon Sep 17 00:00:00 2001 From: veerlosar Date: Sun, 13 Oct 2019 12:12:11 +0200 Subject: [PATCH 3/3] changed the order --- sklearn/multiclass.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sklearn/multiclass.py b/sklearn/multiclass.py index 00c5c07b9032d..78ab288a6aecf 100644 --- a/sklearn/multiclass.py +++ b/sklearn/multiclass.py @@ -194,8 +194,8 @@ class OneVsRestClassifier(MultiOutputMixin, ClassifierMixin, ... ]) >>> y = np.array([0, 0, 1, 1, 2, 2]) >>> clf = OneVsRestClassifier(SVC()).fit(X, y) - >>> clf.predict([[-19, -20], [-5, 5], [9, 9]]) - array([2, 1, 0]) + >>> clf.predict([[-19, -20], [9, 9], [-5, 5]]) + array([2, 0, 1]) """ def __init__(self, estimator, n_jobs=None):