嘗試調整動量（momentum）

在這個練習中，你的目標是找出最佳的動量（momentum），讓最佳化器能在 20 步內找到下列非凸函式 \(x^{4} + x^{3} - 5x^{2}\) 的最小值。你會嘗試兩個不同的動量數值。此問題中的學習率固定為 0.01。

你已獲得 optimize_and_plot() 函式，它接受 momentum 參數作為輸入。這個函式會執行 SGD 最佳化器的 20 步，並顯示結果。