TeraSpace Den4ikAI commited on
Commit
9247da5
1 Parent(s): 24b0d37

Update README.md (#2)

Browse files

- Update README.md (65dc9c214c32c42cb1dc3af8bc1d5dcbe8a382fb)


Co-authored-by: Denis Petrov <[email protected]>

Files changed (1) hide show
  1. README.md +7 -6
README.md CHANGED
@@ -21,19 +21,20 @@ widget:
21
  # TeraSpace/replica_classification
22
  Сделано на основе [xlm-roberta-base](https://huggingface.co/xlm-roberta-base)
23
 
24
- 1. about_user - реагирует, когда пользователь говорит о себе. Например, "меня зовут андрей"
25
- 2. quest - реагирует на вопрос не относящийся к диалогу, например: "когда родился пушкин" или "когда я стану миллионером"
26
- 3. instruct - реагирует на вопросы, ответ на которые представляет собой инструкцию. Например, "как установить windows, как приготовить борщ"
 
27
  4. about_model - реагирует на вопросы о личности ассистента. Например, "как тебя зовут, ты кто такая"
28
- 5. trouble - реагирует на реплики, где пользователь рассказывает о своих проблемах. Например, "у меня болит зуб, мне проткнули колесо"
29
- 7. dialog - реагирует на диалоговые реплики. Например, "привет"
30
  # Использование
31
  ```python
32
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
33
  import torch
34
 
35
  del_symbs = ["?","!",".",","]
36
- classes = ["dialog","trouble","quest","about_user","about_model","instruction"]
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'