Justus-Jonas commited on
Commit
9bacc2d
1 Parent(s): 72352ce

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -0
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
+ ```