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
Branches
Tags
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.
Please register or to comment