Bug fix
parent
b243b64551
commit
d3a8e1dd1d
2
CNN.py
2
CNN.py
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue