分類法/範例三: Plot classification probability

這個範例的主要目的

  • 使用iris 鳶尾花資料集
  • 測試不同分類器對於涵蓋特定範圍之資料集,分類為那一種鳶尾花的機率
  • 例如:sepal length 為 4cm 而 sepal width 為 3cm時被分類為 versicolor的機率

(一)資料匯入及描述

  • 首先先匯入iris 鳶尾花資料集,使用iris = datasets.load_iris()將資料存入
  • 準備X (特徵資料) 以及 y (目標資料),僅使用兩個特徵方便視覺呈現
  1. import matplotlib.pyplot as plt
  2. import numpy as np
  3. from sklearn.linear_model import LogisticRegression
  4. from sklearn.svm import SVC
  5. from sklearn import datasets
  6. iris = datasets.load_iris()
  7. X = iris.data[:, 0:2] # 僅使用前兩個特徵,方便視覺化呈現
  8. y = iris.target
  9. n_features = X.shape[1]
  • iris為一個dict型別資料,我們可以用以下指令來看一下資料的內容。
  1. for key,value in iris.items() :
  2. try:
  3. print (key,value.shape)
  4. except:
  5. print (key)
顯示 說明
(‘target_names’, (3L,)) 共有三種鳶尾花 setosa, versicolor, virginica
(‘data’, (150L, 4L)) 有150筆資料,共四種特徵
(‘target’, (150L,)) 這150筆資料各是那一種鳶尾花
DESCR 資料之描述
feature_names 四個特徵代表的意義

(二) 分類器的選擇

這個範例選擇了四種分類器,存入一個dict資料中,分別為:

  1. L1 logistic
  2. L2 logistic (OvR)
  3. Linear SVC
  4. L2 logistic (Multinomial)

其中LogisticRegression 並不適合拿來做多目標的分類器,我們可以用結果圖的分類機率來觀察。

  1. C = 1.0
  2. # Create different classifiers. The logistic regression cannot do
  3. # multiclass out of the box.
  4. classifiers = {'L1 logistic': LogisticRegression(C=C, penalty='l1'),
  5. 'L2 logistic (OvR)': LogisticRegression(C=C, penalty='l2'),
  6. 'Linear SVC': SVC(kernel='linear', C=C, probability=True,
  7. random_state=0),
  8. 'L2 logistic (Multinomial)': LogisticRegression(
  9. C=C, solver='lbfgs', multi_class='multinomial'
  10. )}
  11. n_classifiers = len(classifiers)

而接下來為了產生一個包含絕大部份可能的測試矩陣,我們會用到以下指令。

  1. np.linspace(起始, 終止, 數量) 目的為產生等間隔之數據,例如print(np.linspace(1,3,3)) 的結果為 [ 1. 2. 3.],而print(np.linspace(1,3,5))的結果為 [ 1. 1.5 2. 2.5 3. ]
  2. np.meshgrid(xx,yy)則用來產生網格狀座標。
  3. numpy.c_ 為numpy特殊物件,能協助將numpy 陣列連接起來,將程式簡化後,我們用以下範例展示相關函式用法。
  1. xx, yy = np.meshgrid(np.linspace(1,3,3), np.linspace(4,6,3).T)
  2. Xfull = np.c_[xx.ravel(), yy.ravel()]
  3. print('xx= \n%s\n' % xx)
  4. print('yy= \n%s\n' % yy)
  5. print('xx.ravel()= %s\n' % xx.ravel())
  6. print('Xfull= \n%s' % Xfull)

結果顯示如下,我們可以看出Xfull模擬出了一個類似特徵矩陣X, 具備有9筆資料,這九筆資料重現了xx (3種數值變化)及yy(3種數值變化)的所有排列組合。

  1. xx=
  2. [[ 1. 2. 3.]
  3. [ 1. 2. 3.]
  4. [ 1. 2. 3.]]
  5. yy=
  6. [[ 4. 4. 4.]
  7. [ 5. 5. 5.]
  8. [ 6. 6. 6.]]
  9. xx.ravel()= [ 1. 2. 3. 1. 2. 3. 1. 2. 3.]
  10. Xfull=
  11. [[ 1. 4.]
  12. [ 2. 4.]
  13. [ 3. 4.]
  14. [ 1. 5.]
  15. [ 2. 5.]
  16. [ 3. 5.]
  17. [ 1. 6.]
  18. [ 2. 6.]
  19. [ 3. 6.]]

而下面這段程式碼的主要用意,在產生一個網格矩陣,其中xx,yy分別代表著iris資料集的第一及第二個特徵。xx 是3~9之間的100個連續數字,而yy是1~5之間的100個連續數字。用np.meshgrid(xx,yy)np.c_產生出Xfull特徵矩陣,10,000筆資料包含了兩個特徵的所有排列組合。

  1. plt.figure(figsize=(3 * 2, n_classifiers * 2))
  2. plt.subplots_adjust(bottom=.2, top=.95)
  3. xx = np.linspace(3, 9, 100)
  4. yy = np.linspace(1, 5, 100).T
  5. xx, yy = np.meshgrid(xx, yy)
  6. Xfull = np.c_[xx.ravel(), yy.ravel()]

(三) 測試分類器以及畫出機率分佈圖的選擇

接下來的動作

  1. 用迴圈輪過所有的分類器,並計算顯示分類成功率
  2. Xfull(10000x2矩陣)傳入 classifier.predict_proba()得到probas(10000x3矩陣)。這裏的probas矩陣是10000種不同的特徵排列組合所形成的數據,被分類到三種iris 鳶尾花的可能性。
  3. 利用reshape((100,100))將10000筆資料排列成二維矩陣,並將機率用影像的方式呈現出來
  1. #若在ipython notebook (Jupyter) 裏執行,則可以將下列這行的井號移除
  2. %matplotlib inline
  3. #原範例沒有下列這行,這是為了讓圖形顯示更漂亮而新增的
  4. fig = plt.figure(figsize=(12,12), dpi=300)
  5. for index, (name, classifier) in enumerate(classifiers.items()):
  6. #訓練並計算分類成功率
  7. #然而此範例訓練跟測試用相同資料集,並不符合實際狀況。
  8. #建議採用cross_validation的方式才能較正確評估
  9. classifier.fit(X, y)
  10. y_pred = classifier.predict(X)
  11. classif_rate = np.mean(y_pred.ravel() == y.ravel()) * 100
  12. print("classif_rate for %s : %f " % (name, classif_rate))
  13. # View probabilities=
  14. probas = classifier.predict_proba(Xfull)
  15. n_classes = np.unique(y_pred).size
  16. for k in range(n_classes):
  17. plt.subplot(n_classifiers, n_classes, index * n_classes + k + 1)
  18. plt.title("Class %d" % k)
  19. if k == 0:
  20. plt.ylabel(name)
  21. imshow_handle = plt.imshow(probas[:, k].reshape((100, 100)),
  22. extent=(3, 9, 1, 5), origin='lower')
  23. plt.xticks(())
  24. plt.yticks(())
  25. idx = (y_pred == k)
  26. if idx.any():
  27. plt.scatter(X[idx, 0], X[idx, 1], marker='o', c='k')
  28. ax = plt.axes([0.15, 0.04, 0.7, 0.05])
  29. plt.title("Probability")
  30. plt.colorbar(imshow_handle, cax=ax, orientation='horizontal')
  31. plt.show()
  1. classif_rate for L2 logistic (OvR) : 76.666667
  2. classif_rate for L1 logistic : 79.333333
  3. classif_rate for Linear SVC : 82.000000
  4. classif_rate for L2 logistic (Multinomial) : 82.000000

png

(四)完整程式碼

Python source code: plot_classification_probability.py

http://scikit-learn.org/stable/_downloads/plot_classification_probability.py

  1. print(__doc__)
  2. # Author: Alexandre Gramfort <alexandre.gramfort@inria.fr>
  3. # License: BSD 3 clause
  4. import matplotlib.pyplot as plt
  5. import numpy as np
  6. from sklearn.linear_model import LogisticRegression
  7. from sklearn.svm import SVC
  8. from sklearn import datasets
  9. iris = datasets.load_iris()
  10. X = iris.data[:, 0:2] # we only take the first two features for visualization
  11. y = iris.target
  12. n_features = X.shape[1]
  13. C = 1.0
  14. # Create different classifiers. The logistic regression cannot do
  15. # multiclass out of the box.
  16. classifiers = {'L1 logistic': LogisticRegression(C=C, penalty='l1'),
  17. 'L2 logistic (OvR)': LogisticRegression(C=C, penalty='l2'),
  18. 'Linear SVC': SVC(kernel='linear', C=C, probability=True,
  19. random_state=0),
  20. 'L2 logistic (Multinomial)': LogisticRegression(
  21. C=C, solver='lbfgs', multi_class='multinomial'
  22. )}
  23. n_classifiers = len(classifiers)
  24. plt.figure(figsize=(3 * 2, n_classifiers * 2))
  25. plt.subplots_adjust(bottom=.2, top=.95)
  26. xx = np.linspace(3, 9, 100)
  27. yy = np.linspace(1, 5, 100).T
  28. xx, yy = np.meshgrid(xx, yy)
  29. Xfull = np.c_[xx.ravel(), yy.ravel()]
  30. for index, (name, classifier) in enumerate(classifiers.items()):
  31. classifier.fit(X, y)
  32. y_pred = classifier.predict(X)
  33. classif_rate = np.mean(y_pred.ravel() == y.ravel()) * 100
  34. print("classif_rate for %s : %f " % (name, classif_rate))
  35. # View probabilities=
  36. probas = classifier.predict_proba(Xfull)
  37. n_classes = np.unique(y_pred).size
  38. for k in range(n_classes):
  39. plt.subplot(n_classifiers, n_classes, index * n_classes + k + 1)
  40. plt.title("Class %d" % k)
  41. if k == 0:
  42. plt.ylabel(name)
  43. imshow_handle = plt.imshow(probas[:, k].reshape((100, 100)),
  44. extent=(3, 9, 1, 5), origin='lower')
  45. plt.xticks(())
  46. plt.yticks(())
  47. idx = (y_pred == k)
  48. if idx.any():
  49. plt.scatter(X[idx, 0], X[idx, 1], marker='o', c='k')
  50. ax = plt.axes([0.15, 0.04, 0.7, 0.05])
  51. plt.title("Probability")
  52. plt.colorbar(imshow_handle, cax=ax, orientation='horizontal')
  53. plt.show()