Skip to content

LindgeW/DualEncoder4TextSimilarity

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DualEncoder4TextSimilarity

Dual Encoder for Text Similarity and Vector Generation

Requirement:

  • python 3.6 +
  • Pytorch >= 1.5
  • cuda >= 10.0

Training Objectives

  • Cosine Similarity Loss
  • Multiple Negatives Ranking Loss
  • Circle Loss (加权Triplet Loss)

About

Dual Encoder for Text Similarity and Vector Generation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published