Parameter Optimization Tutorial: https://codecraft.tv/courses/tensorflowjs/tensors/optimisation/ Documentation for "Gradient descent (with momentum) optimizer": https://www.tensorflow.org/api_docs/python/tf/keras/optimizers/SGD