Minor fix to infer_pt2.py
Browse files- infer_pt2.py +1 -1
infer_pt2.py
CHANGED
|
@@ -69,7 +69,7 @@ def main() -> None:
|
|
| 69 |
|
| 70 |
if len(embeddings) >= 2:
|
| 71 |
cos = torch.dot(embeddings[0], embeddings[1]).item()
|
| 72 |
-
print(f"\
|
| 73 |
|
| 74 |
|
| 75 |
if __name__ == "__main__":
|
|
|
|
| 69 |
|
| 70 |
if len(embeddings) >= 2:
|
| 71 |
cos = torch.dot(embeddings[0], embeddings[1]).item()
|
| 72 |
+
print(f"\ncosine similarity (sentences 1 & 2): {cos:+.4f}")
|
| 73 |
|
| 74 |
|
| 75 |
if __name__ == "__main__":
|