
302 Глава 9
>>> print(f'MAE при тес тировании: {mae_test:.2f}')
MAE при те стировании: 20821.77
>>> r2_train = r2_score(y_train, y_train_pred)
>>> r2_test =r2_score(y_test, y_test_pred)
>>> print(f'R^2 при обу чении: {r2_train:.2f}')
R^2 при об учении: 0.98
>>> print(f'R^2 при тес тировании: {r2_test:.2f}')
R^2 при те стировании: 0.85
К сожалению, мы видим, что случайный лес склонен к переобучению на обучающих
данных. Тем не менее он все еще может относительно хорошо объяснить взаимосвязь
между целевыми и независимыми переменными ("
2
= 0.85 в тестовом наборе данных).
Для сравнения: линейная модель из предыдущего раздела, обученная