FabrizioMusacchio / Gradient_Descent

This repository contains the code for the blog post on Understanding gradient descent in machine learning. For further details, please refer to this post.

Gradient Descent

This repository contains the code for the blog post on Understanding gradient descent in machine learning. For further details, please refer to this post.

You can run the notebooks in Google Colab or Binder by clicking on one of the buttons below:

One parameter problem:

Two parameter problem:

For reproducibility:

conda create -n gradient_descent -y python=3.9
conda activate gradient_descent
conda install -y mamba
mamba install -y ipykernel numpy matplotlib ipywidgets notebook

Acknowledgement

The main code is based on the blog post "Visualizing the gradient descent method" from scipython.com.