oluwatosin adewumi
commited on
Commit
•
28ed5a4
1
Parent(s):
5187dc7
readme doc edited
Browse files
README.md
CHANGED
@@ -14,23 +14,32 @@ widget:
|
|
14 |
- text: "How I fit chop for here?"
|
15 |
---
|
16 |
|
17 |
-
## DialoGPT_AfriWOZ
|
|
|
18 |
This is a fine-tuned model of DialoGPT (small) on the AfriWOZ dataset. It is intended to be used as a conversational system in Nigeria Pidgin English language.
|
19 |
The dataset it's trained on is limited in scope, as it covers only certain domains such as restaurants, hotel, taxi, and booking.
|
20 |
-
|
|
|
|
|
21 |
* Generation example from an interactive environment:
|
|
|
22 |
|Role | Response |
|
23 |
-
|
24 |
|User | I hear say restaurant dey here. |
|
25 |
-
|Bot |
|
26 |
|User | Abeg you fit tell me which kind chop dey? |
|
27 |
-
|Bot |
|
28 |
|User | You do well. Thank you. |
|
29 |
-
|Bot | |
|
|
|
30 |
Please find the information about preprocessing, training and full details of the DialoGPT in the [original DialoGPT repository](https://github.com/microsoft/DialoGPT)
|
|
|
31 |
The paper for this work can be found on arXiv: [https://arxiv.org/pdf/2204.08083.pdf](https://arxiv.org/pdf/2204.08083.pdf)
|
|
|
32 |
### How to use
|
|
|
33 |
Now we are ready to try out how the model works as a chatting partner!
|
|
|
34 |
```python
|
35 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
36 |
import torch
|
|
|
14 |
- text: "How I fit chop for here?"
|
15 |
---
|
16 |
|
17 |
+
## DialoGPT_AfriWOZ (Pidgin)
|
18 |
+
|
19 |
This is a fine-tuned model of DialoGPT (small) on the AfriWOZ dataset. It is intended to be used as a conversational system in Nigeria Pidgin English language.
|
20 |
The dataset it's trained on is limited in scope, as it covers only certain domains such as restaurants, hotel, taxi, and booking.
|
21 |
+
|
22 |
+
The perplexity achieved on the validation set is 38.52.
|
23 |
+
|
24 |
* Generation example from an interactive environment:
|
25 |
+
|
26 |
|Role | Response |
|
27 |
+
|------|------------|
|
28 |
|User | I hear say restaurant dey here. |
|
29 |
+
|Bot | I dey find restaurant. |
|
30 |
|User | Abeg you fit tell me which kind chop dey? |
|
31 |
+
|Bot | I no dey sure right now. |
|
32 |
|User | You do well. Thank you. |
|
33 |
+
|Bot | I dey find restauarant. |
|
34 |
+
|
35 |
Please find the information about preprocessing, training and full details of the DialoGPT in the [original DialoGPT repository](https://github.com/microsoft/DialoGPT)
|
36 |
+
|
37 |
The paper for this work can be found on arXiv: [https://arxiv.org/pdf/2204.08083.pdf](https://arxiv.org/pdf/2204.08083.pdf)
|
38 |
+
|
39 |
### How to use
|
40 |
+
|
41 |
Now we are ready to try out how the model works as a chatting partner!
|
42 |
+
|
43 |
```python
|
44 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
45 |
import torch
|