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',
|