https://machinelearningmodels.org/building-a-decision-tree-classifier-in-scikit-learn/