尝试不同的学习率

在本练习中，您的目标是在 10 步内，让优化器找到非凸函数 \(x^{4} + x^{3} - 5x^{2}\) 的最小值，从而确定一个合适的学习率。

您将尝试 3 个不同的学习率值。针对这个问题，请在 0.001 到 0.1 之间选择学习率。

我们已提供 optimize_and_plot() 函数，其第 1 个参数为学习率。该函数会运行 10 步 SGD 优化器并显示结果。