注意:单击此处https://urlify.cn/aERj2e下载完整的示例代码,或通过Binder在浏览器中运行此示例
绘制不同分类器的分类概率。我们使用三分类的数据集,并使用支持向量分类器,具有一对多(One-Vs-Rest)或多项式设置的L1和L2惩罚的逻辑回归以及高斯过程分类对数据集进行分类。
线性SVC在默认情况下不是概率分类器,但在此示例中启用了内置校准选项(built-in calibration option)(
sphx_glr_plot_classification_probability_001
输出:
probability=True
)。
具有一对多(One-Vs-Rest)的逻辑回归不是立即可用(out of the box)的多分类器,结果与其他分类器相比,在分离第2类和第3类时会有更多的麻烦。
![4e953cac60be71056cbdf6911e2f5b7c.png](https://i-blog.csdnimg.cn/blog_migrate/bdc04ab3cf9dcbe629522e834f689eed.png)
Accuracy (train) for L1 logistic: 82.7%Accuracy (train) for L2 logistic (Multinomial): 82.7%Accuracy (train) for L2 logistic (OvR): 79.3%Accuracy (train) for Linear SVC: 82.0%Accuracy (train) for GPC: 82.7%
print(__doc__)# 作者: Alexandre Gramfort # 许可证: BSD 3 clauseimport matplotlib.pyplot as pltimport numpy as npfrom sklearn.metrics import accuracy_scorefrom sklearn.linear_model import LogisticRegressionfrom sklearn.svm import SVCfrom sklearn.gaussian_process import GaussianProcessClassifierfrom sklearn.gaussian_process.kernels import RBFfrom sklearn import datasetsiris = datasets.load_iris()X = iris.data[:, 0:2] # 我们仅用前两个特征进行可视化y = iris.targetn_features = X.shape[1]C = 10kernel = 1.0 * RBF([1.0, 1.0]) # 对于GPC# 创建不同的分类器classifiers = { 'L1 logistic': LogisticRegression(C=C, penalty='l1', solver='saga', multi_class='multinomial', max_iter=10000), 'L2 logistic (Multinomial)': LogisticRegression(C=C, penalty='l2', solver='saga', multi_class='multinomial', max_iter=10000), 'L2 logistic (OvR)': LogisticRegression(C=C, penalty='l2', solver='saga', multi_class='ovr', max_iter=10000), 'Linear SVC': SVC(kernel='linear', C=C, probability=True, random_state=0), 'GPC': GaussianProcessClassifier(kernel)}n_classifiers = len(classifiers)plt.figure(figsize=(3 * 2, n_classifiers * 2))plt.subplots_adjust(bottom=.2, top=.95)xx = np.linspace(3, 9, 100)yy = np.linspace(1, 5, 100).Txx, yy = np.meshgrid(xx, yy)Xfull = np.c_[xx.ravel(), yy.ravel()]for index, (name, classifier) in enumerate(classifiers.items()): classifier.fit(X, y) y_pred = classifier.predict(X) accuracy = accuracy_score(y, y_pred) print("Accuracy (train) for %s: %0.1f%% " % (name, accuracy * 100)) # 查看概率 probas = classifier.predict_proba(Xfull) n_classes = np.unique(y_pred).size for k in range(n_classes): plt.subplot(n_classifiers, n_classes, index * n_classes + k + 1) plt.title("Class %d" % k) if k == 0: plt.ylabel(name) imshow_handle = plt.imshow(probas[:, k].reshape((100, 100)), extent=(3, 9, 1, 5), origin='lower') plt.xticks(()) plt.yticks(()) idx = (y_pred == k) if idx.any(): plt.scatter(X[idx, 0], X[idx, 1], marker='o', c='w', edgecolor='k')ax = plt.axes([0.15, 0.04, 0.7, 0.05])plt.title("Probability")plt.colorbar(imshow_handle, cax=ax, orientation='horizontal')plt.show()
脚本的总运行时间:(0分钟1.188秒)
估计的内存使用量: 45 MB
![8653b1f8e077cc1d4f5061420d4e7128.png](https://i-blog.csdnimg.cn/blog_migrate/7bedf6dc51f85cc1f7dc2fb2b43dc97a.png)
下载Python源代码: plot_classification_probability.py
下载Jupyter notebook源代码: plot_classification_probability.ipynb
由Sphinx-Gallery生成的画廊
文壹由“伴编辑器”提供技术支持
☆☆☆为方便大家查阅,小编已将scikit-learn学习路线专栏 文章统一整理到公众号底部菜单栏,同步更新中,关注公众号,点击左下方“系列文章”,如图:欢迎大家和我一起沿着scikit-learn文档这条路线,一起巩固机器学习算法基础。(添加微信:mthler,备注:sklearn学习,一起进【sklearn机器学习进步群】开启打怪升级的学习之旅。)