From 1ec023951862a5318527fecb2ff1d45305045543 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 12 Jan 2021 11:10:59 +0800 Subject: Typo correct in evaluate function --- models/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/model.py b/models/model.py index b343f86..c6f6e57 100644 --- a/models/model.py +++ b/models/model.py @@ -273,7 +273,7 @@ class Model: (labels_p, _, views_p, features_p) = probe_samples_c.values() views_p = np.asarray(views_p) for (v_p_i, view_p) in enumerate(probe_views_meta): - probe_view_mask = (views_p == view_g) + probe_view_mask = (views_p == view_p) f_p = features_p[probe_view_mask] y_p = labels_p[probe_view_mask] # Euclidean distance -- cgit v1.2.3