This repository contains the training and prediction of neural network code for square of a number
The detailed explanation of this notebook can be found in medium article: https://towardsdatascience.com/the-simplest-neural-network-understanding-the-non-linearity-10846d7d0141
Note: If notebook is not getting rendered, then open https://nbviewer.jupyter.org/ and paste link: https://github.com/tseth92/NeuralNet_Square/blob/master/neural_network_square.ipynb. Then you can open notebook in colab from the button provided there.