codewithzichao / SimCSE-Pytorch

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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) img.png

About

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


Languages

Language:Python 100.0%