PySpark Decision Tree Regressor leaf nodes as input for Linear Regression Model to make Enhanced Decision Tree Regressor

Question

This post is about implementing a similar concept to this post. Essentially I want to create an Enhanced Decision Tree Regressor by creating a Decision Tree Regressor (DTR)