Bom dia,
Para função trazer o mesmo resultado alterei uma linha e squared, que ficou assim:
from sklearn.metrics import mean_squared_error, mean_absolute_error, r2_score
import numpy as np
def calcular_metricas_regressao2(y_test, y_pred):
mse = mean_squared_error(y_test, y_pred)
rmse = np.sqrt(mse)
mae = mean_absolute_error(y_test, y_pred)
r2 = r2_score(y_test, y_pred)
metricas = {
'Raiz do Erro Quadrático Médio': round(rmse, 4),
'Erro Absoluto Médio': round(mae, 4),
'R2 Score': round(r2, 4)
}
return metricas
Conte com o apoio da comunidade Alura na sua jornada. Abraços e bons estudos!