From dcd66a8a8a6a6d9fae2b37a6722ea39831734de3 Mon Sep 17 00:00:00 2001 From: Noel Dawe Date: Tue, 15 Jan 2013 13:30:14 +0100 Subject: [PATCH] fix class ordering in two-class --- examples/ensemble/plot_adaboost_twoclass.py | 2 +- sklearn/ensemble/weight_boosting.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/ensemble/plot_adaboost_twoclass.py b/examples/ensemble/plot_adaboost_twoclass.py index bb3f18b9ebb..bee02156bc5 100644 --- a/examples/ensemble/plot_adaboost_twoclass.py +++ b/examples/ensemble/plot_adaboost_twoclass.py @@ -58,7 +58,7 @@ pl.legend(loc='upper right') pl.xlabel("Decision Boundary") # Plot the class probabilities -class_proba = bdt.predict_proba(X)[:, 0] +class_proba = bdt.predict_proba(X)[:, -1] pl.subplot(132) for i, n, c in zip(xrange(2), class_names, plot_colors): pl.hist(class_proba[y == i], diff --git a/sklearn/ensemble/weight_boosting.py b/sklearn/ensemble/weight_boosting.py index 1a0e616101c..1237fefc10e 100644 --- a/sklearn/ensemble/weight_boosting.py +++ b/sklearn/ensemble/weight_boosting.py @@ -695,7 +695,7 @@ class AdaBoostClassifier(BaseWeightBoosting, ClassifierMixin): if i == n_estimators: break - purities = estimator.predict_proba(X)[:, 0] + purities = estimator.predict_proba(X)[:, -1] norm += weight if output is None: