https://aiappworld.com/tutorial/visualization-machine-learning-in-python/