├── .gitignore ├── README.md └── src ├── data ├── data_process.py └── dataset.py ├── main.py └── model ├── predict.py ├── simcse_sup.py ├── simcse_unsup.py ├── train_sup.py └── train_unsup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seanzhang-zhichen/simcse-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seanzhang-zhichen/simcse-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /src/data/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seanzhang-zhichen/simcse-pytorch/HEAD/src/data/data_process.py -------------------------------------------------------------------------------- /src/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seanzhang-zhichen/simcse-pytorch/HEAD/src/data/dataset.py -------------------------------------------------------------------------------- /src/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seanzhang-zhichen/simcse-pytorch/HEAD/src/main.py -------------------------------------------------------------------------------- /src/model/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seanzhang-zhichen/simcse-pytorch/HEAD/src/model/predict.py -------------------------------------------------------------------------------- /src/model/simcse_sup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seanzhang-zhichen/simcse-pytorch/HEAD/src/model/simcse_sup.py -------------------------------------------------------------------------------- /src/model/simcse_unsup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seanzhang-zhichen/simcse-pytorch/HEAD/src/model/simcse_unsup.py -------------------------------------------------------------------------------- /src/model/train_sup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seanzhang-zhichen/simcse-pytorch/HEAD/src/model/train_sup.py -------------------------------------------------------------------------------- /src/model/train_unsup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seanzhang-zhichen/simcse-pytorch/HEAD/src/model/train_unsup.py --------------------------------------------------------------------------------