พยากรณ์บนชุดทดสอบ

เมื่อแบ่งข้อมูลเป็นชุดฝึกและชุดทดสอบแบบสุ่มแล้ว ให้ใช้ฟังก์ชัน lm() เช่นเดียวกับแบบฝึกหัดแรก เพื่อ fit โมเดลกับชุดฝึกแทนที่จะใช้ข้อมูลทั้งหมด ทบทวนการใช้ formula interface กับฟังก์ชัน linear regression เพื่อ fit โมเดลที่กำหนดตัวแปรเป้าหมาย โดยใช้ตัวแปรอื่นทั้งหมดในชุดข้อมูลเป็น predictor:

mod <- lm(y ~ ., training_data)

ใช้ฟังก์ชัน predict() เพื่อพยากรณ์ผลจากโมเดลบนข้อมูลใหม่ได้ ชุดข้อมูลใหม่ต้องมีคอลัมน์ครบเหมือนกับชุดฝึก แต่ลำดับและค่าของข้อมูลอาจแตกต่างกันได้ ในขั้นตอนนี้ แทนที่จะพยากรณ์ซ้ำบนชุดฝึก ให้พยากรณ์บนชุดทดสอบที่ไม่ได้ใช้ในการฝึกโมเดล เพื่อนำไปคำนวณ out-of-sample error ในแบบฝึกหัดถัดไป:

p <- predict(model, new_data)