codewithzichao / SimCSE-Pytorch

中文数据集下SimCSE+ESimCSE的实现

SimCSE Inplemention

SimCSE在中文上无监督 + 有监督 pytorch版

论文：https://arxiv.org/pdf/2104.08821.pdf

1.数据 SNS-B

directory data/SNS-B/

2.how to run

python train.py

3.Result (un-supervised)
Look at the column of STS-B, SimCSE-BERT
top spearman corrcoef at default setup parameter: 0.7535 (paper 0.768)