This is my code:
regr = DecisionTreeRegressor(max_depth = 2) regr.fit(X_train, y_train) y_pred = regr.predict(X_test) mse = mean_squared_error(y_test, y_pr