TEST:    MSE: 16.726, RMSE: 4.090, MAE: 2.336, ME: -0.594, gaussian_nll: -0.657, laplacian_nll: -0.164, loss: -0.657,