IDEA-CCNL
/

Erlangshen-TCBert-1.3B-Sentence-Embedding-Chinese

Model card Files Files and versions Community

idea-teacher commited on Jun 21, 2023

Commit

3c9007b

•

1 Parent(s): 1f835ed

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -182,7 +182,7 @@ with torch.no_grad():
     test_representation = torch.mean(training_outputs.hidden_states[-1].squeeze(), dim=0)
 # Calculate similarity scores
-similarity_score = cos(training_input, test_input)
 ```
 ## 引用 Citation

     test_representation = torch.mean(training_outputs.hidden_states[-1].squeeze(), dim=0)
 # Calculate similarity scores
+similarity_score = cos(training_representation, test_representation)
 ```
 ## 引用 Citation