#Building the Decision Tree Model on our dataset from sklearn.tree import DecisionTreeRegressor DT_model = DecisionTreeRegressor(max_depth=5).fit(X_train,Y_train) DT_predict = DT_model.predict(X_test) #Predictions on Testing data print(DT_predict)