from sklearn.ensemble import RandomForestRegressor clf = RandomForestRegressor(max_depth=2, random_state=0) clf.fit(X, y) print(clf.predict([[0, 0, 0, 0]]))