開始使用免費開始

從機率到混淆矩陣

相反地,假設你想要非常確定模型能把所有水雷都正確辨識為水雷。這時你可能會把預測門檻設為 0.10,而不是 0.90。

將機率切成預測類別,接著計算混淆矩陣的程式碼範式,已在本章的練習 7 示範過。

本練習屬於課程

使用 R 的 caret 進行 Machine Learning

檢視課程

練習說明

  • 使用 ifelse() 建立字元向量 m_or_r:當 p 大於 0.1 時為正類別 "M",否則為負類別 "R"
  • m_or_r 轉為因子 p_class,其層級需與 test[["Class"]] 相同。
  • 使用 confusionMatrix() 建立混淆矩陣,並傳入 p_classtest 資料集的 "Class" 欄。

動手互動練習

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

# If p exceeds threshold of 0.1, M else R: m_or_r


# Convert to factor: p_class


# Create confusion matrix
編輯並執行程式碼