Implementing decision tree regression