Diwank Tomer
commited on
Commit
·
ab3c427
1
Parent(s):
71993a6
Update README.md
Browse files
README.md
CHANGED
|
@@ -9,19 +9,17 @@ license: mit
|
|
| 9 |
Takes two sentences as inputs (one previous and one current utterance of a dialog). The previous sentence can be an empty string if this is the first utterance of a speaker in a dialog. **Outputs one of 11 labels**:
|
| 10 |
|
| 11 |
```python
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
(10, 'ask_yes_no')
|
| 24 |
-
]
|
| 25 |
```
|
| 26 |
|
| 27 |
## Example:
|
|
@@ -31,7 +29,7 @@ from simpletransformers.classification import (
|
|
| 31 |
ClassificationModel, ClassificationArgs
|
| 32 |
)
|
| 33 |
|
| 34 |
-
model = ClassificationModel("deberta", "diwank/
|
| 35 |
convert_to_label = lambda n: [
|
| 36 |
['acknowledge',
|
| 37 |
'answer',
|
|
|
|
| 9 |
Takes two sentences as inputs (one previous and one current utterance of a dialog). The previous sentence can be an empty string if this is the first utterance of a speaker in a dialog. **Outputs one of 11 labels**:
|
| 10 |
|
| 11 |
```python
|
| 12 |
+
(0, 'acknowledge')
|
| 13 |
+
(1, 'answer')
|
| 14 |
+
(2, 'backchannel')
|
| 15 |
+
(3, 'reply_yes')
|
| 16 |
+
(4, 'exclaim')
|
| 17 |
+
(5, 'say')
|
| 18 |
+
(6, 'reply_no')
|
| 19 |
+
(7, 'hold')
|
| 20 |
+
(8, 'ask')
|
| 21 |
+
(9, 'intent')
|
| 22 |
+
(10, 'ask_yes_no')
|
|
|
|
|
|
|
| 23 |
```
|
| 24 |
|
| 25 |
## Example:
|
|
|
|
| 29 |
ClassificationModel, ClassificationArgs
|
| 30 |
)
|
| 31 |
|
| 32 |
+
model = ClassificationModel("deberta", "diwank/silicone-deberta-pair")
|
| 33 |
convert_to_label = lambda n: [
|
| 34 |
['acknowledge',
|
| 35 |
'answer',
|