https://medium.com/@alains/data-science-a-step-by-step-guide-to-training-a-model-on-a-new-dataset-af30916a7b67