calculate train set distances to the reference embedding

nguyenvulong · Dec 7, 2023 · 7f0ff30 · 7f0ff30
1 parent bd3ecf7
commit 7f0ff30
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/oc_classifier.py b/oc_classifier.py
@@ -192,6 +192,8 @@ def create_reference_embedding2(model, dataloader, device):
     for emb in total_embeddings:
         distance = F.pairwise_distance(reference_embedding, emb, p=2)
         total_distances.append(distance)
+        with open("distances.txt", "a") as f:
+            f.write(f"{float(distance)}\n")
     threshold = torch.max(torch.stack(total_distances))
 
     # save the reference embedding and threshold to a file