fix correct display of mse

This commit is contained in:
2021-05-18 16:11:51 +02:00
parent ff0a416e26
commit 43c77768ee

View File

@@ -80,7 +80,7 @@ def predict(model, data, grid_h, grid_w, comparison_data=pd.DataFrame()):
err = normalize_df(part_distance_out) - normalize_df(comparison_data_dist)
err = err * err
plot_party_distances(err)
plt.title(f'distance squared error, with mse={str(np.nanmean(err.to_numpy())):.2}')
plt.title(f'distance squared error, with mse={np.nanmean(err.to_numpy()):.2f}')
plt.show()