├── README.md ├── data ├── STS-B │ ├── dev.txt │ ├── test.txt │ └── train.txt ├── cmnli_sup_train_data.csv ├── ocnli_sup_train_data.csv ├── ocnli_unsup_train_data.txt └── stsb_unsup_train_data.txt ├── dataset.py ├── eval_bert.py ├── image └── train.jpg ├── make_train_data.py ├── model.py ├── output └── tensorboard │ ├── cmnli_sup-bsz-64-dropout-0.1 │ └── events.out.tfevents.1646205592.d3460abe-c7da-4a39-8635-a19bfb027a17.24461.0 │ ├── ocnli_sup-bsz-64-dropout-0.1 │ └── events.out.tfevents.1646204557.d3460abe-c7da-4a39-8635-a19bfb027a17.21945.0 │ ├── ocnli_unsup-bsz-32-dropout-0.1 │ └── events.out.tfevents.1646202323.d3460abe-c7da-4a39-8635-a19bfb027a17.17142.0 │ └── sts_unsup-bsz-32-dropout-0.2 │ └── events.out.tfevents.1646201073.d3460abe-c7da-4a39-8635-a19bfb027a17.14154.0 ├── requirements.txt ├── script ├── run_sup_train.sh └── run_unsup_train.sh └── train_sts.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/README.md -------------------------------------------------------------------------------- /data/STS-B/dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/data/STS-B/dev.txt -------------------------------------------------------------------------------- /data/STS-B/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/data/STS-B/test.txt -------------------------------------------------------------------------------- /data/STS-B/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/data/STS-B/train.txt -------------------------------------------------------------------------------- /data/cmnli_sup_train_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/data/cmnli_sup_train_data.csv -------------------------------------------------------------------------------- /data/ocnli_sup_train_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/data/ocnli_sup_train_data.csv -------------------------------------------------------------------------------- /data/ocnli_unsup_train_data.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/data/ocnli_unsup_train_data.txt -------------------------------------------------------------------------------- /data/stsb_unsup_train_data.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/data/stsb_unsup_train_data.txt -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/dataset.py -------------------------------------------------------------------------------- /eval_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/eval_bert.py -------------------------------------------------------------------------------- /image/train.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/image/train.jpg -------------------------------------------------------------------------------- /make_train_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/make_train_data.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/model.py -------------------------------------------------------------------------------- /output/tensorboard/cmnli_sup-bsz-64-dropout-0.1/events.out.tfevents.1646205592.d3460abe-c7da-4a39-8635-a19bfb027a17.24461.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/output/tensorboard/cmnli_sup-bsz-64-dropout-0.1/events.out.tfevents.1646205592.d3460abe-c7da-4a39-8635-a19bfb027a17.24461.0 -------------------------------------------------------------------------------- /output/tensorboard/ocnli_sup-bsz-64-dropout-0.1/events.out.tfevents.1646204557.d3460abe-c7da-4a39-8635-a19bfb027a17.21945.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/output/tensorboard/ocnli_sup-bsz-64-dropout-0.1/events.out.tfevents.1646204557.d3460abe-c7da-4a39-8635-a19bfb027a17.21945.0 -------------------------------------------------------------------------------- /output/tensorboard/ocnli_unsup-bsz-32-dropout-0.1/events.out.tfevents.1646202323.d3460abe-c7da-4a39-8635-a19bfb027a17.17142.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/output/tensorboard/ocnli_unsup-bsz-32-dropout-0.1/events.out.tfevents.1646202323.d3460abe-c7da-4a39-8635-a19bfb027a17.17142.0 -------------------------------------------------------------------------------- /output/tensorboard/sts_unsup-bsz-32-dropout-0.2/events.out.tfevents.1646201073.d3460abe-c7da-4a39-8635-a19bfb027a17.14154.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/output/tensorboard/sts_unsup-bsz-32-dropout-0.2/events.out.tfevents.1646201073.d3460abe-c7da-4a39-8635-a19bfb027a17.14154.0 -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch==1.10 2 | transformers==3.1 3 | numpy 4 | loguru 5 | tensorboard 6 | scipy -------------------------------------------------------------------------------- /script/run_sup_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/script/run_sup_train.sh -------------------------------------------------------------------------------- /script/run_unsup_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/script/run_unsup_train.sh -------------------------------------------------------------------------------- /train_sts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jjljkjljk/SimCSE-Chinese/HEAD/train_sts.py --------------------------------------------------------------------------------