Instructions to use YuxinJiang/unsup-promcse-bert-base-uncased with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use YuxinJiang/unsup-promcse-bert-base-uncased with Transformers:
# Load model directly from transformers import AutoTokenizer, BertForCL tokenizer = AutoTokenizer.from_pretrained("YuxinJiang/unsup-promcse-bert-base-uncased") model = BertForCL.from_pretrained("YuxinJiang/unsup-promcse-bert-base-uncased", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
4bd9790
1
Parent(s): 78131e0
Update README.md
Browse files
README.md
CHANGED
|
@@ -1,35 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# PromCSE: Improved Universal Sentence Embeddings with Prompt-based Contrastive Learning and Energy-based Learning
|
| 2 |
[](https://colab.research.google.com/drive/1lanXViJzbmGM1bwm8AflNUKmrvDidg_3?usp=sharing)
|
| 3 |
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
**************************** **Updates** ****************************
|
| 7 |
-
* 2023/4/5: We released our sentence embedding [python package](#getting-started).
|
| 8 |
-
* 2022/3/3: We released a simple [colab notebook](https://colab.research.google.com/drive/1lanXViJzbmGM1bwm8AflNUKmrvDidg_3?usp=sharing) for a quick start!
|
| 9 |
-
* 2022/1/8: We released our model checkpoints on [huggingface](https://huggingface.co/YuxinJiang).
|
| 10 |
-
* 2022/10/9: We released the second verson of [our paper](https://arxiv.org/pdf/2203.06875v2.pdf). Check it out!
|
| 11 |
-
* 2022/10/6: Our paper has been accepted to [**EMNLP 2022**](https://2022.emnlp.org/).
|
| 12 |
-
* 2022/3/14: We released the first verson of [our paper](https://arxiv.org/pdf/2203.06875v1.pdf). Check it out!
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
## Quick Links
|
| 18 |
-
- [Overview](#overview)
|
| 19 |
-
- [Model List](#model-list)
|
| 20 |
-
- [Usage](#usage)
|
| 21 |
-
- [Train PromCSE](#train-promcse)
|
| 22 |
-
- [Setups](#setups)
|
| 23 |
-
- [Evaluation](#evaluation)
|
| 24 |
-
- [Training](#training)
|
| 25 |
-
- [Citation](#citation)
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
## Overview
|
| 30 |
-
<img src="https://github.com/YJiangcm/PromCSE/blob/master/figure/overview.jpg" width="700" height="320">
|
| 31 |
-
|
| 32 |
|
|
|
|
| 33 |
|
| 34 |
## Model List
|
| 35 |
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
---
|
| 4 |
# PromCSE: Improved Universal Sentence Embeddings with Prompt-based Contrastive Learning and Energy-based Learning
|
| 5 |
[](https://colab.research.google.com/drive/1lanXViJzbmGM1bwm8AflNUKmrvDidg_3?usp=sharing)
|
| 6 |
|
| 7 |
+
arXiv link: https://arxiv.org/abs/2203.06875v2
|
| 8 |
+
Published in [**EMNLP 2022**](https://2022.emnlp.org/)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
+
Our code is modified based on [SimCSE](https://github.com/princeton-nlp/SimCSE) and [P-tuning v2](https://github.com/THUDM/P-tuning-v2/). Here we would like to sincerely thank them for their excellent works.
|
| 11 |
|
| 12 |
## Model List
|
| 13 |
|