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)
|
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}")
|
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_rmse = test_rmse
|
||||||
best_params = params
|
best_params = params
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue