開始使用免費開始

建立簡單的決策樹

loans 資料集包含 11,312 位隨機抽樣、向美國的點對點借貸公司 Lending Club 申請並獲得核貸的人。

你將使用決策樹,根據申請時的申貸金額和信用評分,學習這些貸款最終結果(已還款或違約)的模式。

接著,比較當申請者信用良好與信用不佳時,樹模型的預測有何不同。

loansgood_creditbad_credit 資料集都已為你載入。

本練習屬於課程

R 監督式學習:分類

檢視課程

練習說明

  • 載入 rpart 套件。
  • 使用 rpart() 函式擬合一個決策樹模型。
    • 在第一個引數提供 R formula,指定 outcomeloan_amountcredit_score 的函數。
    • 先不要動 control 引數。(之後你會再深入學習!)
  • 使用 predict() 和得到的貸款模型來預測 good_credit 申請者的結果。將 type 引數設為 "class",以預測結果的類別。
  • bad_credit 申請者做相同的預測。

動手互動練習

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

# Load the rpart package


# Build a lending model predicting loan outcome versus loan amount and credit score
loan_model <- rpart(___, data = ___, method = "___", control = rpart.control(cp = 0))

# Make a prediction for someone with good credit
predict(___, ___, type = "___")

# Make a prediction for someone with bad credit
編輯並執行程式碼