Update README.md
Browse files
README.md
CHANGED
@@ -16,11 +16,27 @@ This model is a fine-tuned version of [roberta-base](https://huggingface.co/robe
|
|
16 |
|
17 |
## Model description
|
18 |
|
19 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
20 |
|
21 |
## Intended uses & limitations
|
22 |
|
23 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
24 |
|
25 |
## Training and evaluation data
|
26 |
|
|
|
16 |
|
17 |
## Model description
|
18 |
|
19 |
+
Custom data generated labeling text according to these five categories.
|
20 |
+
Five categories represent the five essential intents of a user for the ACTS scenario.
|
21 |
+
|
22 |
+
- Connect : Greetings and introduction with the student
|
23 |
+
- Pump : Asking the student for information
|
24 |
+
- Inform : Providing information to the student
|
25 |
+
- Feedback : Praising the student (positive feedback) or informing the student they are not on the right path (negative feedback)
|
26 |
+
- None : Not related to scenario
|
27 |
+
|
28 |
+
Takes a user input of string text and classifies it according to one of five categories.
|
29 |
|
30 |
## Intended uses & limitations
|
31 |
|
32 |
+
|
33 |
+
from transformers import pipeline
|
34 |
+
classifier = pipeline("text-classification",model="mp6kv/main_intent_test")
|
35 |
+
|
36 |
+
|
37 |
+
output = classifier("great job, you're getting it!")
|
38 |
+
output[0]['score']
|
39 |
+
output[0]['label']
|
40 |
|
41 |
## Training and evaluation data
|
42 |
|