開始使用免費開始

計算混淆矩陣

正如你在影片中學到的,羅吉斯迴歸模型會產生兩種類型的預測:一種是連續值的預測(機率),另一種是類別預測。在 wells 資料集的例子中,類別是二元的離散類別。

在前一個練習中,你已經計算出以機率表示的連續值預測。這個練習中,你會用那些數值,將 wells_test 樣本中的每筆觀測指定為一個類別。最後,你會用混淆矩陣來描述模型。

計算好的預測 predictionwells_test 已經載入到你的工作區。

本練習屬於課程

Generalized Linear Models in Python

檢視課程

動手互動練習

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

# Define the cutoff
cutoff = ____

# Compute class predictions: y_prediction
y_prediction = np.where(____ > ____, 1, 0)
編輯並執行程式碼