diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 11:10:59 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 11:10:59 +0800 |
| commit | 1ec023951862a5318527fecb2ff1d45305045543 (patch) | |
| tree | 125bdfd7c6f2f1978cd536c1063b94a00ce5c049 /models/model.py | |
| parent | 35a7218c7b445409829d3725b10c3b7d3cb51541 (diff) | |
Typo correct in evaluate function
Diffstat (limited to 'models/model.py')
| -rw-r--r-- | models/model.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 |