Hồi quy tuyến tính với hàm mất mát MSE L=n1∑i(y^i−yi)2, trong đó y^=Xw và X đã thêm cột bias 1. Gradient: ∇L=n2X⊤(Xw−y). Thực hiện đúng steps bước gradient descent:
w←w−η∇L
với learning rate η và vector khởi tạo w cho trong input. In vector w cuối cùng (gồm bias rồi các trọng số đặc trưng).
Dòng đầu: n m. Dòng 2: steps và η. Tiếp theo n dòng dữ liệu (m đặc trưng + y). Dòng cuối: m+1 số khởi tạo w (bias trước).
1≤n≤200; 1≤m≤5; 0≤steps≤1000; 0<η≤1.
Một dòng m+1 số thực là w sau steps bước (2 chữ số thập phân).
Ví dụ:
Đầu vào:
2 1
1 0.1
0 0
2 2
0 0
Đầu ra:
0.20 0.40
Giải thích:
Đang tải editor...