Update README.md (#2)
Browse files- Update README.md (65dc9c214c32c42cb1dc3af8bc1d5dcbe8a382fb)
Co-authored-by: Denis Petrov <[email protected]>
README.md
CHANGED
@@ -21,19 +21,20 @@ widget:
|
|
21 |
# TeraSpace/replica_classification
|
22 |
Сделано на основе [xlm-roberta-base](https://huggingface.co/xlm-roberta-base)
|
23 |
|
24 |
-
|
25 |
-
|
26 |
-
|
|
|
27 |
4. about_model - реагирует на вопросы о личности ассистента. Например, "как тебя зовут, ты кто такая"
|
28 |
-
5.
|
29 |
-
|
30 |
# Использование
|
31 |
```python
|
32 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
33 |
import torch
|
34 |
|
35 |
del_symbs = ["?","!",".",","]
|
36 |
-
classes = ["dialog","trouble","
|
37 |
|
38 |
device = torch.device("cuda")
|
39 |
model_name = 'TeraSpace/replica_classification'
|
|
|
21 |
# TeraSpace/replica_classification
|
22 |
Сделано на основе [xlm-roberta-base](https://huggingface.co/xlm-roberta-base)
|
23 |
|
24 |
+
0. dialog - реагирует на диалоговые реплики. Например, "привет"
|
25 |
+
1. trouble - реагирует на реплики, где пользователь рассказывает о своих проблемах. Например, "у меня болит зуб, мне проткнули колесо"
|
26 |
+
2. question - реагирует на вопрос не относящийся к диалогу, например: "когда родился пушкин" или "когда я стану миллионером"
|
27 |
+
3. about_user - реагирует, когда пользователь говорит о себе. Например, "меня зовут андрей"
|
28 |
4. about_model - реагирует на вопросы о личности ассистента. Например, "как тебя зовут, ты кто такая"
|
29 |
+
5. instruct - реагирует на вопросы, ответ на которые представляет собой инструкцию. Например, "как установить windows, как приготовить борщ"
|
30 |
+
|
31 |
# Использование
|
32 |
```python
|
33 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
34 |
import torch
|
35 |
|
36 |
del_symbs = ["?","!",".",","]
|
37 |
+
classes = ["dialog","trouble","question","about_user","about_model","instruct"]
|
38 |
|
39 |
device = torch.device("cuda")
|
40 |
model_name = 'TeraSpace/replica_classification'
|