remove error

cosine_sim_btw_embeddings_of_same_semantics.py

@@ -22,7 +22,7 @@ Examples:
 
     Example 1-A simple example
         >>> cos_similarity_metrics = evaluate.load("ahnyeonchan/cosine_sim_btw_embeddings_of_same_semantics")
-        >>> results = accuracy_metric.compute(references=[[1.0, 1.0], [0.0, 1.0]], predictions=[[1.0, 1.0], [0.0, 1.0]])
+        >>> results = cos_similarity_metrics.compute(references=[[1.0, 1.0], [0.0, 1.0]], predictions=[[1.0, 1.0], [0.0, 1.0]])
         >>> print(results)
         {'cos_similarity': 1.0}
 """
@@ -66,7 +66,7 @@ class CosSim(evaluate.Metric):
             raise NotImplementedError()
         
         cosims = self.cossim(predictions, references)
-        val = torch.mean(cossim).item()
+        val = torch.mean(cosims).item()
         
         return {
             "cos_similarity": float(val)