Justus-Jonas
commited on
Commit
•
9bacc2d
1
Parent(s):
72352ce
Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,23 @@
|
|
1 |
---
|
2 |
license: cc-by-nc-sa-4.0
|
|
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: cc-by-nc-sa-4.0
|
3 |
+
pipeline_tag: conversational
|
4 |
---
|
5 |
+
|
6 |
+
Imaginary Embeddings utilize Curved Contrastive Learning (see paper [Imagination Is All You Need!](https://arxiv.org/pdf/2211.07591.pdf) (ACL 2023)) on [Sentence Transformers](https://sbert.net/) for long-short term dialogue planning and efficient abstract sequence modeling.
|
7 |
+
|
8 |
+
This model uses speaker tokens and was evaluated in the Short-Term planning experiments.
|
9 |
+
|
10 |
+
## Setup
|
11 |
+
```bash
|
12 |
+
python -m pip install imaginaryNLP
|
13 |
+
```
|
14 |
+
|
15 |
+
## Usage
|
16 |
+
```python
|
17 |
+
candidates = ['Want to eat something out ?',
|
18 |
+
'Want to go for a walk ?']
|
19 |
+
|
20 |
+
goal = ' I am hungry.'
|
21 |
+
|
22 |
+
stp.short_term_planning(candidates, goal)
|
23 |
+
```
|