https://adeveloperdiary.github.io/data-science/deep-learning/neural-network-with-softmax-in-python/