update readme.

henryfcb · Aug 11, 2022 · ad5b8f6 · ad5b8f6
1 parent bc49e1d
commit ad5b8f6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -750,11 +750,13 @@ Xu, M. Text2vec: Text to vector toolkit (Version 1.1.2) [Computer software]. htt
 
 BibTeX:
 ```latex
-@software{Xu_Text2vec_Text_to,
-author = {Xu, Ming},
-title = {{Text2vec: Text to vector toolkit}},
-url = {https://github.com/shibing624/text2vec},
-version = {1.1.2}
+@misc{Text2vec,
+  author = {Xu, Ming},
+  title = {Text2vec: Text to vector toolkit},
+  year = {2022},
+  publisher = {GitHub},
+  journal = {GitHub repository},
+  howpublished = {\url{https://github.com/shibing624/text2vec}},
 }
 ```
 

diff --git a/text2vec/cosent_model.py b/text2vec/cosent_model.py
@@ -121,7 +121,7 @@ def train_model(
 
     def calc_loss(self, y_true, y_pred):
         """
-        CoSENT的排序loss，refer：https://kexue.fm/archives/8847
+        矩阵计算batch内的cos loss
         """
         # 1. 取出真实的标签
         y_true = y_true[::2]  # tensor([1, 0, 1]) 真实的标签