- Optimize Rosenbrock function,use Armijo condition to determinate step of GD, about Rosenbrock function https://en.wikipedia.org/wiki/Rosenbrock_function
- Numpy is required, matplotlib and mpl_toolkits are needed for plotting image.
- run
python hw1.py
- You may need to give hw1.py execute access by
chmod +x hw1.py
- You could change the x dimension by changing
self.dimension
- You could change iteration time by changing
self.iter_times