Дерево решений в регрессии

import pandas as pd
from sklearn.tree import DecisionTreeRegressor
from sklearn.metrics import mean_squared_error

for depth in range(1, 6):

----model = DecisionTreeRegressor(random_state=5678, max_depth=depth)