Bài tập | NEU OJ

Cho n quan sát (x_i, y_i) và số fold k (n chia hết cho k). Hãy chia tuần tự (không xáo trộn) thành k fold theo thứ tự đầu vào: fold j gồm các chỉ số j*(n/k) .. j*(n/k) + n/k - 1. Với mỗi fold làm tập test, dùng phần còn lại fit hồi quy tuyến tính đơn biến y = a*x + b bằng np.polyfit(x_train, y_train, 1), sau đó tính MSE trên fold test = mean((y_pred - y_test)^2). In ra k giá trị MSE theo thứ tự fold, cách nhau dấu cách, làm tròn 4 chữ số thập phân, kèm MSE trung bình ở cuối. Tổng cộng k+1 số trên một dòng.

Định dạng đầu vào:
Dòng 1: n k. Dòng 2: n số x. Dòng 3: n số y.
Ràng buộc đầu vào:
2 <= k <= 10; k+1 <= n <= 200; n chia hết cho k
Định dạng đầu ra:
Một dòng: mse_fold0 mse_fold1 ... mse_fold(k-1) mse_avg (4 chữ số).

solution

Đề bài: [Data Science] K-fold CV MSE cho hồi quy đơn biến