開始使用免費開始

混淆矩陣

使用 scikit-learn 的 confusion_matrix() 函式,你可以輕鬆建立分類器的混淆矩陣,並更細緻地了解其效能。它需要兩個引數:測試集的實際標籤 y_test,以及你的預測標籤。

上一題中隨機森林分類器的預測標籤已存於 y_pred,其計算方式如下:

y_pred = clf.predict(X_test)

重要說明:sklearn 在預設情況下是以下列方式計算混淆矩陣:

Screenshot 2019-05-13 05.59.04.png

請注意,座標軸與你在影片中看到的是相反的。指標本身不變,但在解讀表格時要記得這點。

本練習屬於課程

行銷分析:用 Python 預測客戶流失

檢視課程

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# Import confusion_matrix
編輯並執行程式碼