Enhancing Unsupervised Sentence Embeddings via Knowledge-Driven Data Augmentation and Gaussian-Decayed Contrastive Learning
Paper
•
2409.12887
•
Published
We release our model checkpoints in huggingface as listed below:
| Model | Avg. STS |
|---|---|
| aleversn/GCSE-BERT-base | 81.98 |
| aleversn/GCSE-BERT-large | 83.07 |
| aleversn/GCSE-RoBERTa-base | 82.12 |
| aleversn/GCSE-RoBERTa-large | 83.82 |
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("aleversn/GCSE-BERT-base")
model = AutoModel.from_pretrained("aleversn/GCSE-RoBERTa-large")
Base model
FacebookAI/roberta-large