IDEA-CCNL
/

Erlangshen-TCBert-1.3B-Classification-Chinese

Model card Files Files and versions Community

idea-teacher commited on Jun 21, 2023

Commit

c8d57e5

•

1 Parent(s): f7aef8d

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