計算混淆矩陣
正如你在影片中學到的,羅吉斯迴歸模型會產生兩種類型的預測:一種是連續值的預測(機率),另一種是類別預測。在 wells 資料集的例子中,類別是二元的離散類別。
在前一個練習中,你已經計算出以機率表示的連續值預測。這個練習中,你會用那些數值,將 wells_test 樣本中的每筆觀測指定為一個類別。最後,你會用混淆矩陣來描述模型。
計算好的預測 prediction 和 wells_test 已經載入到你的工作區。
本練習屬於課程
Generalized Linear Models in Python
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
# Define the cutoff
cutoff = ____
# Compute class predictions: y_prediction
y_prediction = np.where(____ > ____, 1, 0)