/cs6493-sts

Primary LanguageJupyter Notebook

cs6493-sts

  1. dataset-prepare-v2.ipynb to get wiki dataset and SentEval dataset
  2. simcse-v2.ipynb to train simcse model , simcse-v2.1.ipynb could skip data preparing to save time when execute repeatly
  3. run_prompt_bert.sh for prompt_bert training
  4. run_dcpsce_unsup.sh for dcpsce training