Logistic regression บนชุดข้อมูล Sonar

Machine Learning ด้วย caret ใน R

Max Kuhn

Software Engineer at RStudio and creator of caret

โมเดลการจำแนกประเภท

ตัวแปรเป้าหมายเป็นข้อมูลเชิงหมวดหมู่ (เชิงคุณภาพ)
ตัวอย่าง: สินเชื่อจะผิดนัดชำระหรือไม่?
ยังคงเป็นรูปแบบหนึ่งของ supervised learning
ใช้การแบ่ง train/test เพื่อประเมินประสิทธิภาพ
ใช้ชุดข้อมูล Sonar
เป้าหมาย: แยกแยะหินกับทุ่นระเบิด

ตัวอย่าง: ข้อมูล Sonar

# Load the Sonar dataset
library(mlbench)
data(Sonar)

# Look at the data
Sonar[1:6, c(1:5, 61)]

      V1     V2     V3     V4     V5 Class
1 0.0200 0.0371 0.0428 0.0207 0.0954     R
2 0.0453 0.0523 0.0843 0.0689 0.1183     R
3 0.0262 0.0582 0.1099 0.1083 0.0974     R
4 0.0100 0.0171 0.0623 0.0205 0.0205     R
5 0.0762 0.0666 0.0481 0.0394 0.0590     R
6 0.0286 0.0453 0.0277 0.0174 0.0384     R

การแบ่งข้อมูล

แบ่งข้อมูลแบบสุ่มเป็นชุด training และ test
ใช้การแบ่ง 60/40 แทน 80/20
ชุดข้อมูล Sonar มีขนาดเล็ก การแบ่ง 60/40 จึงให้ชุด test ที่ใหญ่และน่าเชื่อถือกว่า

การแบ่งข้อมูล

# Randomly order the dataset
rows <- sample(nrow(Sonar))
Sonar <- Sonar[rows, ]

# Find row to split on
split <- round(nrow(Sonar) * 0.60)
train <- Sonar[1:split, ]
test <- Sonar[(split + 1):nrow(Sonar), ]

# Confirm test set size
nrow(train) / nrow(Sonar)

0.6009615

มาฝึกกันเถอะ!

Machine Learning ด้วย caret ใน R