Skip to content
Snippets Groups Projects
Commit 5cc50846 authored by Jörg Martin's avatar Jörg Martin
Browse files

True RMSE added

parent 77b46e57
No related branches found
No related tags found
No related merge requests found
......@@ -139,6 +139,8 @@ def collect_metrics(x_y_pairs, seed=0,
epistemic_coverage(not_averaged_predictions, true_y,
average_predictions=True,
normalize_errors=False, noisy_y=False)
true_res = true_y - noneiv_mean
noneiv_metrics['true_rmse'] = np.sqrt(np.mean(scaled_res**2))
# NLL
......@@ -208,7 +210,8 @@ def collect_metrics(x_y_pairs, seed=0,
average_predictions=True,
normalize_errors=False, noisy_y=False)
true_res = true_y - eiv_mean
eiv_metrics['true_rmse'] = np.sqrt(np.mean(scaled_res**2))
# NLL
if scale_outputs:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment