Training a decision tree model