Tối ưu một tham số vector θ∈Rn bằng momentum. Cho gradient hiện tại g∈Rn, vận tốc trước đó v∈Rn, hệ số quán tính β và tốc độ học η. Một bước cập nhật: v′=βv+g,θ′=θ−ηv′
In ra trên dòng 1 vận tốc mới v′ (n số), trên dòng 2 tham số mới θ′ (n số). Tất cả làm tròn 2 chữ số.
Dòng 1: n β η. Dòng 2: θ (n số). Dòng 3: v (n số). Dòng 4: g (n số).
1≤n≤10; 0≤β<1; 0<η≤1.
Dòng 1: v′. Dòng 2: θ′. Đều làm tròn 2 chữ số.
Ví dụ:
Đầu vào:
1 0.9 0.1
1
2
3
Đầu ra:
4.80
0.52
Giải thích:
Đang tải editor...