solution

Đề bài: [Toán cho CNTT] Gradient descent với momentum

Gradient descent có quán tính (momentum)

Cho f(x)=a*x^2+b*x+c, f'(x)=2*a*x+b. Cập nhật momentum:

v <- beta*v + f'(x)
x <- x - lr*v

Khởi tạo v=0, xuất phát x0. Lặp đúng n bước, in x.

Ví dụ

a=1,b=0,c=0,x0=10,lr=0.1,beta=0.9,n=1: v=0.9*0+20=20, x=10-0.1*20=8.000000.

Định dạng đầu vào:
Dòng 1: a b c. Dòng 2: x0 lr beta. Dòng 3: n.
Ràng buộc đầu vào:
a > 0; 0 < lr <= 0.5; 0 <= beta < 1; 1 <= n <= 100000.
Định dạng đầu ra:
Một dòng: x sau n bước. Số thực in với định dạng {:.6f} (6 chữ số thập phân).

Ví dụ:

Đầu vào:

1 0 0
10 0.1 0.9
1

Đầu ra:

8.000000

Giải thích:

v=20, x=10-0.1*20=8.

Đang tải editor...