|
--- |
|
license: cc-by-nc-sa-4.0 |
|
pipeline_tag: conversational |
|
datasets: |
|
- daily_dialog |
|
--- |
|
|
|
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. |
|
|
|
This model uses speaker tokens and was evaluated in the Short-Term planning experiments. |
|
|
|
## Setup |
|
```bash |
|
python -m pip install imaginaryNLP |
|
``` |
|
|
|
## Usage |
|
```python |
|
candidates = ['Want to eat something out ?', |
|
'Want to go for a walk ?'] |
|
|
|
goal = ' I am hungry.' |
|
|
|
stp.short_term_planning(candidates, goal) |
|
``` |
