Skip to content

SimCSE (Simple Contrastive Learning of Sentence Embeddings) Sample Code: Generating Data & Train

License

Notifications You must be signed in to change notification settings

qbxlvnf11/SimCSE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Contents

- SimCSE (Simple Contrastive Learning of Sentence Embeddings) Sample Code

  • Generating Unsupervised & Supervised SimCSE Data
  • SimCSE Train

Generating Unsupervised & Supervised SimCSE Data

python gen_simcse_data.py --data_path {csv data path} --encoding {encoding} --desc_col {anchor column} --category_col {hard negative column} --positive_cols {positive column1, ...} --output_unsupervised {unsupervised train data path} --output_supervised {supervised train data path}

About

SimCSE (Simple Contrastive Learning of Sentence Embeddings) Sample Code: Generating Data & Train

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages