main
Michelle Goeppinger 2025-02-16 19:16:34 +01:00
parent b243b64551
commit d3a8e1dd1d
1 changed files with 1 additions and 1 deletions

2
CNN.py
View File

@ -192,7 +192,7 @@ if __name__ == '__main__':
test_r2 = r2_score(test_labels, test_preds)
print(f"Model: {model_name} Test RMSE: {test_rmse:.4f}, Test MAE: {test_mae:.4f}, Test R²: {test_r2:.4f}")
if test_rmse > best_params_rmse:
if test_rmse < best_params_rmse:
best_params_rmse = test_rmse
best_params = params