Skip to content

NLIDataset + FULL SimCSE + Rewrite Codes

Qumeng Sun requested to merge dev into main

In this update:

  1. The Sup-SimCSE method was introduced, using the NLI-for-simcse.csv database.
  2. Implemented the early stop on phrases two and three.
  3. **Rewrite **some codes, enhance the readability of the code and add comments, and removed some duplicate code.
  4. Delete the PCG parts in the main code.

After this, the sts task's performance improved (around 2.5%) to around 80.1%. The performance on the other two tasks didn't improve a lot.

Merge request reports