Script for training and validating a tree, then storing it in a pkl