From 1ec023951862a5318527fecb2ff1d45305045543 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
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(-)
(limited to 'models')
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