diwank commited on
Commit
3513688
1 Parent(s): 71c0c5d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -0
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
+ ```