Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,22 @@
|
|
1 |
---
|
2 |
license: mit
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: mit
|
3 |
---
|
4 |
+
|
5 |
+
# maptask-deberta-pair
|
6 |
+
Deberta-based Daily MapTask style dialog-act annotations classification model
|
7 |
+
|
8 |
+
## Example
|
9 |
+
|
10 |
+
```python
|
11 |
+
from simpletransformers.classification import (
|
12 |
+
ClassificationModel, ClassificationArgs
|
13 |
+
)
|
14 |
+
|
15 |
+
model = ClassificationModel("deberta", "diwank/maptask-deberta-pair")
|
16 |
+
|
17 |
+
predictions, raw_outputs = model.predict([["Say what is the meaning of life?", "I dont know"]])
|
18 |
+
|
19 |
+
convert_to_label = lambda n: ["acknowledge (0), align (1), check (2), clarify (3), explain (4), instruct (5), query_w (6), query_yn (7), ready (8), reply_n (9), reply_w (10), reply_y (11)".split(', ')[i] for i in n]
|
20 |
+
|
21 |
+
convert_to_label(predictions) # reply_n (9)
|
22 |
+
```
|