Upload 20 files
Browse files- .gitattributes +1 -0
- Dstasets/Emotions_detected.csv +236 -0
- Dstasets/ru-go-emotions-raw.csv +3 -0
- Dstasets/testEmotionDetected.csv +236 -0
- Scripts/EmotionsTest.py +71 -0
- Scripts/RuBert-tiny2-fine-tune-EmotionsDetected/emotiondetected.ipynb +1 -0
- Scripts/TermoMap.py +111 -0
- Versions/Emotions_model-01.zip +3 -0
- Versions/Emotions_model-02.zip +3 -0
- Versions/Emotions_model-03.zip +3 -0
- Versions/Emotions_model-04.zip +3 -0
- config.json +88 -0
- img/Comparison.png +0 -0
- img/PredictedEmotion.png +0 -0
- img/TrueEmotion.png +0 -0
- model.safetensors +3 -0
- special_tokens_map.json +37 -0
- test.ipynb +179 -0
- tokenizer.json +0 -0
- tokenizer_config.json +65 -0
- vocab.txt +0 -0
.gitattributes
CHANGED
@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
RuBert-tiny2-EmotionsDetected/Dstasets/ru-go-emotions-raw.csv filter=lfs diff=lfs merge=lfs -text
|
|
|
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
RuBert-tiny2-EmotionsDetected/Dstasets/ru-go-emotions-raw.csv filter=lfs diff=lfs merge=lfs -text
|
37 |
+
Dstasets/ru-go-emotions-raw.csv filter=lfs diff=lfs merge=lfs -text
|
Dstasets/Emotions_detected.csv
ADDED
@@ -0,0 +1,236 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
text,true emotions,predict emotions,correctly Identified,incorrectly Identified,undefined
|
2 |
+
Ты сделал отличный выбор!, admiration approval,admiration approval,admiration approval,None,
|
3 |
+
Как же это забавно!, amusement joy,amusement joy,amusement joy,None,
|
4 |
+
"Не могу поверить, что ты так поступил.", disapproval disappointment anger,surprise,None,surprise, disappointment disapproval anger
|
5 |
+
Эта ситуация выводит меня из себя!, annoyance anger,neutral disappointment,None,neutral disappointment, annoyance anger
|
6 |
+
Я всегда готов тебе помочь., love caring,neutral caring approval,caring,neutral approval, love
|
7 |
+
"Не понимаю, как это работает.", curiosity confusion,neutral confusion,confusion,neutral, curiosity
|
8 |
+
"Интересно, что произойдет дальше?", curiosity optimism,curiosity surprise,curiosity,surprise, optimism
|
9 |
+
Я так хочу пойти на этот концерт., excitement desire,desire,desire,None, excitement
|
10 |
+
"Не могу поверить, что это случилось.", sadness disappointment,surprise,None,surprise, sadness disappointment
|
11 |
+
Тебе не стоило так делать., disapproval remorse,neutral disapproval,disapproval,neutral, remorse
|
12 |
+
Меня тошнит от этой еды., disgust anger,sadness,None,sadness, disgust anger
|
13 |
+
Какой позор!, embarrassment sadness,embarrassment,embarrassment,None, sadness
|
14 |
+
Это невероятно захватывающе!, joy admiration excitement,excitement,excitement,None, joy admiration
|
15 |
+
Я боюсь потерять тебя., fear sadness,fear,fear,None, sadness
|
16 |
+
Спасибо тебе огромное за помощь., admiration gratitude,gratitude,gratitude,None, admiration
|
17 |
+
Мне так грустно из-за этого., grief sadness,sadness,sadness,None, grief
|
18 |
+
Как же я рад видеть тебя!, joy love,joy excitement,joy,excitement, love
|
19 |
+
Я люблю тебя больше всего на свете., admiration love,love,love,None, admiration
|
20 |
+
Почему у меня трясутся руки перед выступлением?, embarrassment fear nervousness,curiosity,None,curiosity, embarrassment fear nervousness
|
21 |
+
"Все получится, я в этом уверен.", pride optimism,optimism approval,optimism,approval, pride
|
22 |
+
Я горжусь своей командой., admiration pride,admiration pride,admiration pride,None,
|
23 |
+
Теперь все стало ясно., relief realization,neutral approval,None,neutral approval, realization relief
|
24 |
+
Наконец-то это закончилось., relief,joy,None,joy, relief
|
25 |
+
"Жаль, что я так поступил.", embarrassment sadness remorse,remorse sadness,remorse sadness,None, embarrassment
|
26 |
+
"Мне очень грустно, что так вышло.", grief sadness,sadness,sadness,None, grief
|
27 |
+
Это совершенно неожиданно!, surprise curiosity,surprise,surprise,None, curiosity
|
28 |
+
"Мне все равно, что ты думаешь.", neutral,neutral anger,neutral,anger,
|
29 |
+
Я восхищаюсь твоей настойчивостью., admiration pride,admiration,admiration,None, pride
|
30 |
+
Как же весело было на этой вечеринке!, amusement joy,amusement joy excitement,amusement joy,excitement,
|
31 |
+
Это несправедливо! Я так зол., disappointment anger,annoyance disapproval anger,anger,annoyance disapproval, disappointment
|
32 |
+
Опять этот шум! Сколько можно?, annoyance anger,neutral curiosity,None,neutral curiosity, annoyance anger
|
33 |
+
"Ты такой заботливый, спасибо.", gratitude caring,gratitude caring,gratitude caring,None,
|
34 |
+
"Не понимаю, зачем ты это сделал.", confusion disapproval,neutral confusion,confusion,neutral, disapproval
|
35 |
+
"Интересно, почему он так сказал?", curiosity confusion,curiosity surprise,curiosity,surprise, confusion
|
36 |
+
Мне так хочется попробовать этот десерт., desire,desire,desire,None,
|
37 |
+
"Жаль, что мы опоздали на поезд.", sadness disappointment,disappointment,disappointment,None, sadness
|
38 |
+
"Я думал, что ты поступишь иначе.", disappointment disapproval,neutral,None,neutral, disapproval disappointment
|
39 |
+
Это отвратительно! Как ты можешь это есть?, disgust anger,curiosity disgust,disgust,curiosity, anger
|
40 |
+
Как же неловко получилось на встрече., embarrassment sadness,embarrassment,embarrassment,None, sadness
|
41 |
+
Я так взволнован перед поездкой!, excitement optimism,joy excitement,excitement,joy, optimism
|
42 |
+
Мне страшно заходить в темную комнату., fear nervousness,fear,fear,None, nervousness
|
43 |
+
От всего сердца спасибо за помощь., admiration gratitude,gratitude,gratitude,None, admiration
|
44 |
+
Я не могу справиться с этой потерей., grief sadness,disapproval disappointment,None,disappointment disapproval, grief sadness
|
45 |
+
Такое счастье быть здесь с тобой., joy love gratitude,joy neutral,joy,neutral, love gratitude
|
46 |
+
Моя любовь к тебе не имеет границ., admiration love,love,love,None, admiration
|
47 |
+
Я нервничаю перед этим экзаменом., fear nervousness,fear nervousness,fear nervousness,None,
|
48 |
+
У нас есть все шансы на победу., pride optimism,optimism,optimism,None, pride
|
49 |
+
Я горжусь своими достижениями., pride realization,admiration pride,pride,admiration, realization
|
50 |
+
"Теперь я понял, где была ошибка.", relief realization,neutral realization,realization,neutral, relief
|
51 |
+
"Какое облегчение, что все закончилось.", gratitude relief,joy relief,relief,joy, gratitude
|
52 |
+
"Мне жаль, что я обидел тебя.", sadness remorse caring,remorse sadness,remorse sadness,None, caring
|
53 |
+
Меня переполняет грусть из-за этой утраты., grief sadness,sadness,sadness,None, grief
|
54 |
+
Этого я точно не ожидал!, surprise excitement,neutral disapproval,None,neutral disapproval, surprise excitement
|
55 |
+
"Все идет как обычно, ничего нового.", neutral,neutral approval,neutral,approval,
|
56 |
+
Ты выглядишь потрясающе!, admiration love,admiration,admiration,None, love
|
57 |
+
"Эта игра такая смешная, не могу перестать смеяться!", amusement joy,amusement,amusement,None, joy
|
58 |
+
Как же ты мог меня предать?, sadness anger,curiosity,None,curiosity, sadness anger
|
59 |
+
Я так устал от всего этого., annoyance sadness,neutral sadness,sadness,neutral, annoyance
|
60 |
+
"Мне нравится, как ты заботишься о деталях.", admiration caring,love,None,love, admiration caring
|
61 |
+
"Не понимаю, зачем все это нужно.", confusion disapproval,confusion,confusion,None, disapproval
|
62 |
+
Почему здесь так тихо? Это странно., curiosity surprise,annoyance confusion disgust,None,annoyance confusion disgust, curiosity surprise
|
63 |
+
Я мечтаю отправиться в путешествие., desire optimism,desire optimism,desire optimism,None,
|
64 |
+
Как же мне обидно из-за этого провала., grief disappointment,sadness disappointment,disappointment,sadness, grief
|
65 |
+
Это было неуместно., embarrassment disapproval,disapproval,disapproval,None, embarrassment
|
66 |
+
Меня тошнит от одного только вида этого., fear disgust,neutral,None,neutral, fear disgust
|
67 |
+
Эта ошибка была такой неловкой!, embarrassment remorse,embarrassment,embarrassment,None, remorse
|
68 |
+
Это самое захватывающее событие в моей жизни!, joy pride excitement,excitement,excitement,None, joy pride
|
69 |
+
"Я боюсь, что это плохо закончится.", fear nervousness,fear,fear,None, nervousness
|
70 |
+
"Спасибо за то, что всегда поддерживаешь меня.", love gratitude,gratitude,gratitude,None, love
|
71 |
+
Я не могу смириться с этой утратой., grief sadness,disapproval sadness disappointment,sadness,disappointment disapproval, grief
|
72 |
+
Какая же я счастливая!, joy love,joy excitement,joy,excitement, love
|
73 |
+
Ты моя единственная любовь., joy love admiration,love,love,None, joy admiration
|
74 |
+
Я нервничаю перед каждым важным шагом., fear nervousness realization,fear nervousness,fear nervousness,None, realization
|
75 |
+
"Все будет хорошо, я уверен.", joy relief optimism,optimism approval,optimism,approval, joy relief
|
76 |
+
Мне есть чем гордиться после всего этого., pride realization,pride,pride,None, realization
|
77 |
+
"Теперь я понял, что поступил неправильно.", remorse realization,realization,realization,None, remorse
|
78 |
+
"Какое облегчение узнать, что все в порядке.", gratitude relief,neutral caring,None,neutral caring, gratitude relief
|
79 |
+
Мне так стыдно за свои слова., embarrassment remorse,embarrassment,embarrassment,None, remorse
|
80 |
+
Сложно выразить всю эту грусть словами., grief sadness,sadness disappointment,sadness,disappointment, grief
|
81 |
+
"Никогда бы не подумал, что такое возможно!", surprise excitement curiosity,neutral,None,neutral, surprise excitement curiosity
|
82 |
+
"Все идет как обычно, без изменений.", neutral,neutral approval,neutral,approval,
|
83 |
+
Ты вдохновляешь меня своим трудом., admiration pride,admiration,admiration,None, pride
|
84 |
+
Я не могу перестать смеяться над этой шуткой!, amusement joy,amusement,amusement,None, joy
|
85 |
+
Как ты мог так поступить? Я в ярости., disappointment anger,annoyance curiosity anger,anger,annoyance curiosity, disappointment
|
86 |
+
Эти задержки так раздражают!, annoyance anger,annoyance anger,annoyance anger,None,
|
87 |
+
"Я всегда здесь, если тебе понадобится помощь.", love caring,neutral caring,caring,neutral, love
|
88 |
+
"Это так странно, я ничего не понимаю.", surprise confusion,,None,None, surprise confusion
|
89 |
+
"Интересно, что будет, если мы попробуем иначе.", curiosity optimism,neutral curiosity,curiosity,neutral, optimism
|
90 |
+
Мне так хочется съездить на море., desire,desire,desire,None,
|
91 |
+
"Я разочарован, что мы не успели.", sadness disappointment,sadness disappointment,sadness disappointment,None,
|
92 |
+
Это было совершенно недопустимо., disapproval anger,neutral disapproval,disapproval,neutral, anger
|
93 |
+
Мне становится плохо от одной только мысли об этом., fear disgust,sadness disappointment,None,sadness disappointment, fear disgust
|
94 |
+
"Ситуация была очень неловкой, но я справился.", embarrassment relief,remorse embarrassment,embarrassment,remorse, relief
|
95 |
+
"Я чувствую прилив энергии, когда думаю об этом!", excitement optimism,neutral excitement,excitement,neutral, optimism
|
96 |
+
"Я боюсь, что у нас мало времени.", fear nervousness,fear,fear,None, nervousness
|
97 |
+
Я никогда не забуду твою доброту. Спасибо., admiration gratitude,gratitude,gratitude,None, admiration
|
98 |
+
Мне так тяжело после этой утраты., grief sadness,sadness,sadness,None, grief
|
99 |
+
Как же хорошо снова быть вместе!, joy love,joy,joy,None, love
|
100 |
+
Ты для меня самое дорогое в жизни., admiration love,admiration,admiration,None, love
|
101 |
+
Мои ладони потеют перед началом., fear nervousness,fear,fear,None, nervousness
|
102 |
+
"Мы точно сможем достичь цели, я уверен!", pride optimism,optimism approval,optimism,approval, pride
|
103 |
+
"Я горд тем, чего мы достигли.", admiration pride,neutral,None,neutral, admiration pride
|
104 |
+
Теперь все стало на свои места., relief realization,neutral,None,neutral, realization relief
|
105 |
+
"Какое облегчение, что все получилось.", relief,joy neutral relief,relief,joy neutral,
|
106 |
+
"Мне так жаль, что я не оправдал твоих ожиданий.", sadness remorse,remorse sadness,remorse sadness,None,
|
107 |
+
Я чувствую глубокую печаль из-за этой потери., grief sadness,sadness,sadness,None, grief
|
108 |
+
Это было совершенно неожиданно!, surprise excitement,surprise,surprise,None, excitement
|
109 |
+
"Ничего особенного, просто обычный день.", neutral,neutral,neutral,None,
|
110 |
+
"Ты всегда знаешь, как поднять настроение.", admiration caring,neutral,None,neutral, admiration caring
|
111 |
+
Эта шутка была просто блестящей!, amusement pride,amusement admiration,amusement,admiration, pride
|
112 |
+
"Я не могу поверить, что ты так поступил со мной.", sadness anger,surprise,None,surprise, sadness anger
|
113 |
+
Почему это продолжается? Это ужасно раздражает!, annoyance disappointment,annoyance disgust,annoyance,disgust, disappointment
|
114 |
+
Я ценю твою заботу обо мне., gratitude caring,gratitude admiration,gratitude,admiration, caring
|
115 |
+
"Не понимаю, как мы допустили эту ошибку.", confusion remorse,confusion,confusion,None, remorse
|
116 |
+
Почему это происходит именно так? Мне интересно., curiosity confusion,curiosity,curiosity,None, confusion
|
117 |
+
Я мечтаю посетить эту страну., excitement desire,desire,desire,None, excitement
|
118 |
+
"Я разочарован результатом, но ничего не поделаешь.", disappointment realization,disappointment,disappointment,None, realization
|
119 |
+
Тебе следовало быть внимательнее., disappointment disapproval,neutral caring,None,neutral caring, disapproval disappointment
|
120 |
+
Это выглядит отвратительно. Я не могу это есть., disgust anger,disgust disapproval,disgust,disapproval, anger
|
121 |
+
Моя ошибка была очень неловкой., embarrassment remorse,embarrassment,embarrassment,None, remorse
|
122 |
+
"Я так взволнован, что наконец-то попробую это.", excitement,excitement,excitement,None,
|
123 |
+
"Я боюсь, что меня не примут в команду.", fear sadness,fear,fear,None, sadness
|
124 |
+
"Спасибо, ты настоящий друг.", love gratitude,gratitude,gratitude,None, love
|
125 |
+
Мне так больно из-за потери., grief sadness,sadness disappointment,sadness,disappointment, grief
|
126 |
+
Я счастлив просто быть рядом с тобой., joy love,joy caring,joy,caring, love
|
127 |
+
Ты моя опора и вдохновение., admiration love pride,admiration,admiration,None, pride love
|
128 |
+
Я нервничаю перед этим испытанием., fear nervousness,fear nervousness,fear nervousness,None,
|
129 |
+
У нас есть все шансы на успех!, joy optimism,optimism,optimism,None, joy
|
130 |
+
"Я горжусь, что смог завершить этот проект.", joy pride,admiration pride,pride,admiration, joy
|
131 |
+
"Теперь я понял, почему все произошло именно так.", curiosity realization,neutral realization,realization,neutral, curiosity
|
132 |
+
Облегчение — это самое подходящее слово для описания., gratitude relief,neutral approval,None,neutral approval, gratitude relief
|
133 |
+
"Прости меня, я очень сожалею о своих словах.", sadness remorse,remorse sadness,remorse sadness,None,
|
134 |
+
"Мне так грустно, что мы больше не вместе.", grief sadness,sadness,sadness,None, grief
|
135 |
+
Такого я точно не ожидал!, surprise shock,surprise disapproval,surprise,disapproval, shock
|
136 |
+
"Все спокойно, ничего нового.", neutral,joy approval,None,joy approval, neutral
|
137 |
+
"Ты всегда знаешь, как вдохновить.", admiration joy,admiration neutral approval,admiration,neutral approval, joy
|
138 |
+
"Это было так смешно, что у меня слезы на глазах.", amusement joy,amusement,amusement,None, joy
|
139 |
+
"Ты причинил мне огромную боль, я не могу это простить.", grief anger,remorse sadness,None,remorse sadness, grief anger
|
140 |
+
Почему я должен постоянно это терпеть?, frustration annoyance,neutral curiosity confusion,None,neutral curiosity confusion, annoyance frustration
|
141 |
+
Твоя забота делает мой день лучше., gratitude caring,caring,caring,None, gratitude
|
142 |
+
"Я не понимаю, зачем это вообще нужно.", confusion disapproval,confusion,confusion,None, disapproval
|
143 |
+
"Мне интересно, что скрывается за этой дверью.", curiosity surprise,surprise,surprise,None, curiosity
|
144 |
+
Я мечтаю об этой машине с самого детства., nostalgia desire,neutral love desire,desire,neutral love, nostalgia
|
145 |
+
"Как же больно осознавать, что я подвел их.", disappointment remorse,sadness disappointment,disappointment,sadness, remorse
|
146 |
+
Это было недопустимо и обидно., disapproval anger,disapproval,disapproval,None, anger
|
147 |
+
Этот запах вызывает у меня отвращение., annoyance disgust,disgust,disgust,None, annoyance
|
148 |
+
Та ситуация была слишком неловкой для всех нас., embarrassment regret,embarrassment,embarrassment,None, regret
|
149 |
+
"Я просто не могу дождаться, когда это начнется!", excitement anticipation,curiosity excitement,excitement,curiosity, anticipation
|
150 |
+
"Я боюсь, что не успею вовремя.", fear nervousness,fear,fear,None, nervousness
|
151 |
+
"Спасибо за все, что ты для меня делаешь.", love gratitude,gratitude,gratitude,None, love
|
152 |
+
Моя потеря невыносима., grief despair,neutral disappointment,None,neutral disappointment, grief despair
|
153 |
+
"Я так рад, что ты со мной.", joy love,joy,joy,None, love
|
154 |
+
"Ты единственный, кто делает меня счастливым.", love,joy neutral,None,joy neutral, love
|
155 |
+
"У меня дрожь по телу, я так волнуюсь.", fear nervousness,fear nervousness,fear nervousness,None,
|
156 |
+
"Все будет замечательно, мы справимся.", pride optimism,admiration optimism,optimism,admiration, pride
|
157 |
+
"Я горжусь тем, что стал частью этого.", pride,admiration pride,pride,admiration,
|
158 |
+
"Теперь я понял, как это исправить.", pride realization,neutral realization,realization,neutral, pride
|
159 |
+
"Слава Богу, это закончилось.", relief,gratitude,None,gratitude, relief
|
160 |
+
"Я сильно жалею о том, что сделал.", sadness remorse,remorse sadness,remorse sadness,None,
|
161 |
+
Мне так грустно из-за всего этого., grief sadness,sadness,sadness,None, grief
|
162 |
+
Этого я точно не ожидал — какая неожиданность!, surprise excitement,,None,None, surprise excitement
|
163 |
+
"Ничего особенного, просто обычный разговор.", neutral,neutral,neutral,None,
|
164 |
+
"Этот фильм такой смешной, я смеялся весь вечер!", amusement joy,amusement,amusement,None, joy
|
165 |
+
"Это видео просто уморительное, его хочется смотреть снова и снова.", amusement curiosity excitement,desire,None,desire, amusement curiosity excitement
|
166 |
+
Почему эти ошибки продолжаются? Это так раздражает!, annoyance disappointment anger,annoyance anger,annoyance anger,None, disappointment
|
167 |
+
"Меня раздражает, что приходится объяснять одно и то же снова и снова.", frustration annoyance disapproval,annoyance,annoyance,None, frustration disapproval
|
168 |
+
"Ты сделал замечательную работу, продолжай в том же духе.", admiration pride approval,admiration,admiration,None, pride approval
|
169 |
+
"Твой подход заслуживает не только уважения, но и признания.", admiration approval,admiration caring,admiration,caring, approval
|
170 |
+
"Я всегда готов поддержать тебя, если понадобится помощь.", love gratitude caring,optimism caring approval,caring,approval optimism, love gratitude
|
171 |
+
"Спасибо, что заботишься обо мне и всегда рядом.", approval caring gratitude love,gratitude caring,gratitude caring,None, love approval
|
172 |
+
"Я не понимаю, почему всё так сложно устроено.", frustration confusion,neutral confusion,confusion,neutral, frustration
|
173 |
+
Почему это так сложно? Я ничего не понимаю., annoyance confusion disappointment,confusion,confusion,None, annoyance disappointment
|
174 |
+
"Интересно, почему ты решил сделать именно так?", curiosity realization,curiosity surprise,curiosity,surprise, realization
|
175 |
+
"Мне очень любопытно, как это всё работает.", curiosity confusion excitement,curiosity,curiosity,None, confusion excitement
|
176 |
+
Я так хочу попробовать этот новый десерт!, joy curiosity desire,excitement desire,desire,excitement, joy curiosity
|
177 |
+
Как же хочется отправиться в путешествие этим летом., excitement desire optimism,curiosity desire,desire,curiosity, excitement optimism
|
178 |
+
"Этот запах вызывает у меня отвращение, я даже не могу дышать.", annoyance disgust anger,fear disgust,disgust,fear, annoyance anger
|
179 |
+
Поведение этого человека вызывает искреннее отвращение., annoyance disgust disapproval,disgust,disgust,None, annoyance disapproval
|
180 |
+
"Неожиданно увидеть тебя здесь, это так удивительно!", joy surprise,admiration surprise excitement,surprise,admiration excitement, joy
|
181 |
+
"Как неожиданно! Никогда бы не подумал, что это возможно.", surprise excitement realization,surprise,surprise,None, excitement realization
|
182 |
+
"Всё прошло совершенно обычно, без сюрпризов.", neutral,neutral,neutral,None,
|
183 |
+
Сегодняшний день ничем не отличался от остальных., neutral disappointment,neutral disapproval,neutral,disapproval, disappointment
|
184 |
+
"Этот комикс не только забавный, но и очень умный.", amusement curiosity approval,amusement admiration,amusement,admiration, curiosity approval
|
185 |
+
"Меня просто бесит, что ты не слушаешь!", annoyance disapproval anger,annoyance anger,annoyance anger,None, disapproval
|
186 |
+
"Я искренне рад за тебя, ты отлично справился.", joy pride approval,admiration joy,joy,admiration, pride approval
|
187 |
+
"Спасибо за твою заботу, это так важно для меня.", love gratitude caring,gratitude,gratitude,None, love caring
|
188 |
+
"Мне сложно понять, что ты имеешь в виду.", frustration curiosity confusion,confusion,confusion,None, frustration curiosity
|
189 |
+
"Мне бы хотелось узнать больше о том, как ты это сделал.", admiration curiosity desire,love curiosity desire,curiosity desire,love, admiration
|
190 |
+
"Я мечтаю о времени, когда смогу путешествовать без ограничений.", pride desire optimism,neutral desire,desire,neutral, pride optimism
|
191 |
+
"Эта ситуация была настолько неловкой, что мне до сих пор стыдно.", embarrassment regret remorse,embarrassment disgust,embarrassment,disgust, regret remorse
|
192 |
+
"Я рад, что всё закончилось именно так.", joy gratitude relief,gratitude joy,gratitude joy,None, relief
|
193 |
+
Это решение вызывает у меня смешанные чувства: радость и тревогу., joy fear nervousness,joy,joy,None, fear nervousness
|
194 |
+
"Ничего особенного, просто ещё один день.", neutral,neutral,neutral,None,
|
195 |
+
"Ты великолепно справился с этой задачей, мне это нравится.", admiration approval,admiration,admiration,None, approval
|
196 |
+
Этот подход выглядит разумным и эффективным., realization approval,admiration approval,approval,admiration, realization
|
197 |
+
"Молодец, ты действительно нашёл правильное решение.", pride approval,gratitude admiration approval,approval,gratitude admiration, pride
|
198 |
+
"Я поддерживаю твою идею, она звучит убедительно.", curiosity approval,admiration approval,approval,admiration, curiosity
|
199 |
+
"Хорошо, что ты подошёл к этому вопросу так серьёзно.", pride caring approval,neutral approval,approval,neutral, pride caring
|
200 |
+
"Этот запах настолько отвратителен, что мне стало плохо.", annoyance disgust,disgust,disgust,None, annoyance
|
201 |
+
Его действия вызывают у меня только отвращение., disgust disapproval anger,disgust,disgust,None, disapproval anger
|
202 |
+
"Эта еда выглядит отвратительно, я не могу её попробовать.", disgust disappointment,disgust disapproval,disgust,disapproval, disappointment
|
203 |
+
"Ситуация настолько неприятна, что мне трудно находиться здесь.", disgust nervousness,sadness disappointment,None,sadness disappointment, disgust nervousness
|
204 |
+
"Его слова звучали так мерзко, что я не мог их выносить.", frustration disgust anger,annoyance,None,annoyance, frustration disgust anger
|
205 |
+
"Всё идёт как обычно, никаких изменений.", neutral,neutral approval,neutral,approval,
|
206 |
+
"Ничего интересного не произошло, всё осталось по-прежнему.", neutral disappointment,realization,None,realization, neutral disappointment
|
207 |
+
"Просто обычный день, без особых событий.", neutral,neutral,neutral,None,
|
208 |
+
"Всё стабильно, ничего нового.", neutral,neutral approval,neutral,approval,
|
209 |
+
Вчерашняя встреча прошла довольно обыденно., neutral,neutral approval,neutral,approval,
|
210 |
+
"Я не понимаю, почему этот процесс такой сложный.", frustration confusion,confusion,confusion,None, frustration
|
211 |
+
Мне сложно разобраться в этой инструкции., confusion disapproval,realization,None,realization, confusion disapproval
|
212 |
+
"Я не понимаю, зачем они так поступают.", confusion disappointment,confusion,confusion,None, disappointment
|
213 |
+
"Мне интересно, как это устройство работает.", curiosity,curiosity excitement surprise,curiosity,surprise excitement,
|
214 |
+
Хотел бы узнать больше об этом явлении., curiosity desire,love curiosity desire,curiosity desire,love,
|
215 |
+
Этот эксперимент вызывает у меня много вопросов., curiosity realization,neutral,None,neutral, curiosity realization
|
216 |
+
Почему бы нам не попробовать сделать это иначе?, curiosity optimism,neutral,None,neutral, curiosity optimism
|
217 |
+
Я хотел бы узнать больше о причинах их выбора., curiosity caring,curiosity desire,curiosity,desire, caring
|
218 |
+
"Я бы хотел, чтобы у нас был шанс попробовать эту стратегию.", desire optimism,desire,desire,None, optimism
|
219 |
+
"Хочется вернуться к этой задаче, чтобы улучшить результат.", pride desire,desire optimism,desire,optimism, pride
|
220 |
+
Как же я хочу добиться лучшего понимания этой темы., curiosity desire,neutral desire,desire,neutral, curiosity
|
221 |
+
"Мне неприятно видеть такое поведение, оно недопустимо.", disgust disapproval,annoyance,None,annoyance, disgust disapproval
|
222 |
+
Этот результат вызывает у меня серьёзные сомнения., disappointment disapproval,fear sadness nervousness,None,fear sadness nervousness, disapproval disappointment
|
223 |
+
"Я не могу согласиться с этим решением, оно кажется ошибочным.", frustration disapproval,disapproval,disapproval,None, frustration
|
224 |
+
Эта идея совершенно противоречит нашим принципам., disapproval anger,disapproval,disapproval,None, anger
|
225 |
+
"Не могу повери��ь, что это решение было принято.", confusion disapproval,neutral surprise,None,neutral surprise, confusion disapproval
|
226 |
+
"Неприятно вспоминать эту ситуацию, она была унизительной.", embarrassment remorse,annoyance,None,annoyance, embarrassment remorse
|
227 |
+
"Эта ошибка была настолько очевидной, что мне стало неловко.", embarrassment disapproval,embarrassment,embarrassment,None, disapproval
|
228 |
+
Я всё ещё чувствую неловкость из-за своего поведения., embarrassment regret,embarrassment,embarrassment,None, regret
|
229 |
+
"Приятно видеть, как ты стараешься в работе.", caring approval,admiration,None,admiration, approval caring
|
230 |
+
"Хорошо, что ты сделал первый шаг, это важно.", pride approval,neutral approval,approval,neutral, pride
|
231 |
+
Этот документ выглядит аккуратно и профессионально., approval,admiration approval,approval,admiration,
|
232 |
+
"Мне нравится, как ты организовал процесс.", realization approval,love,None,love, approval realization
|
233 |
+
"Отличная работа, продолжай в том же духе.", optimism approval,admiration,None,admiration, approval optimism
|
234 |
+
"Этот шум действительно раздражает, невозможно сосредоточиться.", frustration annoyance,neutral annoyance,annoyance,neutral, frustration
|
235 |
+
Почему ты не слушаешь? Это уже надоело., annoyance disappointment,annoyance sadness disappointment,annoyance disappointment,sadness,
|
236 |
+
"Кажется, что я постоянно сталкиваюсь с этими мелкими проблемами.", annoyance disapproval,neutral,None,neutral, annoyance disapproval
|
Dstasets/ru-go-emotions-raw.csv
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8d148a7f8c61a260ec66e2dd55d0cbc3ed3ae6df608a7ca0ed57751c9abee5c0
|
3 |
+
size 71457187
|
Dstasets/testEmotionDetected.csv
ADDED
@@ -0,0 +1,236 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
text,emotions
|
2 |
+
"Ты сделал отличный выбор!", approval admiration
|
3 |
+
"Как же это забавно!", amusement joy
|
4 |
+
"Не могу поверить, что ты так поступил.", anger disappointment disapproval
|
5 |
+
"Эта ситуация выводит меня из себя!", annoyance anger
|
6 |
+
"Я всегда готов тебе помочь.", caring love
|
7 |
+
"Не понимаю, как это работает.", confusion curiosity
|
8 |
+
"Интересно, что произойдет дальше?", curiosity optimism
|
9 |
+
"Я так хочу пойти на этот концерт.", desire excitement
|
10 |
+
"Не могу поверить, что это случилось.", disappointment sadness
|
11 |
+
"Тебе не стоило так делать.", disapproval remorse
|
12 |
+
"Меня тошнит от этой еды.", disgust anger
|
13 |
+
"Какой позор!", embarrassment sadness
|
14 |
+
"Это невероятно захватывающе!", excitement joy admiration
|
15 |
+
"Я боюсь потерять тебя.", fear sadness
|
16 |
+
"Спасибо тебе огромное за помощь.", gratitude admiration
|
17 |
+
"Мне так грустно из-за этого.", grief sadness
|
18 |
+
"Как же я рад видеть тебя!", joy love
|
19 |
+
"Я люблю тебя больше всего на свете.", love admiration
|
20 |
+
"Почему у меня трясутся руки перед выступлением?", nervousness fear embarrassment
|
21 |
+
"Все получится, я в этом уверен.", optimism pride
|
22 |
+
"Я горжусь своей командой.", pride admiration
|
23 |
+
"Теперь все стало ясно.", realization relief
|
24 |
+
"Наконец-то это закончилось.", relief
|
25 |
+
"Жаль, что я так поступил.", remorse embarrassment sadness
|
26 |
+
"Мне очень грустно, что так вышло.", sadness grief
|
27 |
+
"Это совершенно неожиданно!", surprise curiosity
|
28 |
+
"Мне все равно, что ты думаешь.", neutral
|
29 |
+
"Я восхищаюсь твоей настойчивостью.", admiration pride
|
30 |
+
"Как же весело было на этой вечеринке!", amusement joy
|
31 |
+
"Это несправедливо! Я так зол.", anger disappointment
|
32 |
+
"Опять этот шум! Сколько можно?", annoyance anger
|
33 |
+
"Ты такой заботливый, спасибо.", caring gratitude
|
34 |
+
"Не понимаю, зачем ты это сделал.", confusion disapproval
|
35 |
+
"Интересно, почему он так сказал?", curiosity confusion
|
36 |
+
"Мне так хочется попробовать этот десерт.", desire
|
37 |
+
"Жаль, что мы опоздали на поезд.", disappointment sadness
|
38 |
+
"Я думал, что ты поступишь иначе.", disapproval disappointment
|
39 |
+
"Это отвратительно! Как ты можешь это есть?", disgust anger
|
40 |
+
"Как же неловко получилось на встрече.", embarrassment sadness
|
41 |
+
"Я так взволнован перед поездкой!", excitement optimism
|
42 |
+
"Мне страшно заходить в темную комнату.", fear nervousness
|
43 |
+
"От всего сердца спасибо за помощь.", gratitude admiration
|
44 |
+
"Я не могу справиться с этой потерей.", grief sadness
|
45 |
+
"Такое счастье быть здесь с тобой.", joy love gratitude
|
46 |
+
"Моя любовь к тебе не имеет границ.", love admiration
|
47 |
+
"Я нервничаю перед этим экзаменом.", nervousness fear
|
48 |
+
"У нас есть все шансы на победу.", optimism pride
|
49 |
+
"Я горжусь своими достижениями.", pride realization
|
50 |
+
"Теперь я понял, где была ошибка.", realization relief
|
51 |
+
"Какое облегчение, что все закончилось.", relief gratitude
|
52 |
+
"Мне жаль, что я обидел тебя.", remorse sadness caring
|
53 |
+
"Меня переполняет грусть из-за этой утраты.", sadness grief
|
54 |
+
"Этого я точно не ожидал!", surprise excitement
|
55 |
+
"Все идет как обычно, ничего нового.", neutral
|
56 |
+
"Ты выглядишь потрясающе!", admiration love
|
57 |
+
"Эта игра такая смешная, не могу перестать смеяться!", amusement joy
|
58 |
+
"Как же ты мог меня предать?", anger sadness
|
59 |
+
"Я так устал от всего этого.", annoyance sadness
|
60 |
+
"Мне нравится, как ты заботишься о деталях.", caring admiration
|
61 |
+
"Не понимаю, зачем все это нужно.", confusion disapproval
|
62 |
+
"Почему здесь так тихо? Это странно.", curiosity surprise
|
63 |
+
"Я мечтаю отправиться в путешествие.", desire optimism
|
64 |
+
"Как же мне обидно из-за этого провала.", disappointment grief
|
65 |
+
"Это было неуместно.", disapproval embarrassment
|
66 |
+
"Меня тошнит от одного только вида этого.", disgust fear
|
67 |
+
"Эта ошибка была такой неловкой!", embarrassment remorse
|
68 |
+
"Это самое захватывающее событие в моей жизни!", excitement pride joy
|
69 |
+
"Я боюсь, что это плохо закончится.", fear nervousness
|
70 |
+
"Спасибо за то, что всегда поддерживаешь меня.", gratitude love
|
71 |
+
"Я не могу смириться с этой утратой.", grief sadness
|
72 |
+
"Какая же я счастливая!", joy love
|
73 |
+
"Ты моя единственная любовь.", love joy admiration
|
74 |
+
"Я нервничаю перед каждым важным шагом.", nervousness fear realization
|
75 |
+
"Все будет хорошо, я уверен.", optimism relief joy
|
76 |
+
"Мне есть чем гордиться после всего этого.", pride realization
|
77 |
+
"Теперь я понял, что поступил неправильно.", realization remorse
|
78 |
+
"Какое облегчение узнать, что все в порядке.", relief gratitude
|
79 |
+
"Мне так стыдно за свои слова.", remorse embarrassment
|
80 |
+
"Сложно выразить всю эту грусть словами.", sadness grief
|
81 |
+
"Никогда бы не подумал, что такое возможно!", surprise curiosity excitement
|
82 |
+
"Все идет как обычно, без изменений.", neutral
|
83 |
+
"Ты вдохновляешь меня своим трудом.", admiration pride
|
84 |
+
"Я не могу перестать смеяться над этой шуткой!", amusement joy
|
85 |
+
"Как ты мог так поступить? Я в ярости.", anger disappointment
|
86 |
+
"Эти задержки так раздражают!", annoyance anger
|
87 |
+
"Я всегда здесь, если тебе понадобится помощь.", caring love
|
88 |
+
"Это так странно, я ничего не понимаю.", confusion surprise
|
89 |
+
"Интересно, что будет, если мы попробуем иначе.", curiosity optimism
|
90 |
+
"Мне так хочется съездить на море.", desire
|
91 |
+
"Я разочарован, что мы не успели.", disappointment sadness
|
92 |
+
"Это было совершенно недопустимо.", disapproval anger
|
93 |
+
"Мне становится плохо от одной только мысли об этом.", disgust fear
|
94 |
+
"Ситуация была очень неловкой, но я справился.", embarrassment relief
|
95 |
+
"Я чувствую прилив энергии, когда думаю об этом!", excitement optimism
|
96 |
+
"Я боюсь, что у нас мало времени.", fear nervousness
|
97 |
+
"Я никогда не забуду твою доброту. Спасибо.", gratitude admiration
|
98 |
+
"Мне так тяжело после этой утраты.", grief sadness
|
99 |
+
"Как же хорошо снова быть вместе!", joy love
|
100 |
+
"Ты для меня самое дорогое в жизни.", love admiration
|
101 |
+
"Мои ладони потеют перед началом.", nervousness fear
|
102 |
+
"Мы точно сможем достичь цели, я уверен!", optimism pride
|
103 |
+
"Я горд тем, чего мы достигли.", pride admiration
|
104 |
+
"Теперь все стало на свои места.", realization relief
|
105 |
+
"Какое облегчение, что все получилось.", relief
|
106 |
+
"Мне так жаль, что я не оправдал твоих ожиданий.", remorse sadness
|
107 |
+
"Я чувствую глубокую печаль из-за этой потери.", sadness grief
|
108 |
+
"Это было совершенно неожиданно!", surprise excitement
|
109 |
+
"Ничего особенного, просто обычный день.", neutral
|
110 |
+
"Ты всегда знаешь, как поднять настроение.", admiration caring
|
111 |
+
"Эта шутка была просто блестящей!", amusement pride
|
112 |
+
"Я не могу поверить, что ты так поступил со мной.", anger sadness
|
113 |
+
"Почему это продолжается? Это ужасно раздражает!", annoyance disappointment
|
114 |
+
"Я ценю твою заботу обо мне.", caring gratitude
|
115 |
+
"Не понимаю, как мы допустили эту ошибку.", confusion remorse
|
116 |
+
"Почему это происходит именно так? Мне интересно.", curiosity confusion
|
117 |
+
"Я мечтаю посетить эту страну.", desire excitement
|
118 |
+
"Я разочарован результатом, но ничего не поделаешь.", disappointment realization
|
119 |
+
"Тебе следовало быть внимательнее.", disapproval disappointment
|
120 |
+
"Это выглядит отвратительно. Я не могу это есть.", disgust anger
|
121 |
+
"Моя ошибка была очень неловкой.", embarrassment remorse
|
122 |
+
"Я так взволнован, что наконец-то попробую это.", excitement
|
123 |
+
"Я боюсь, что меня не примут в команду.", fear sadness
|
124 |
+
"Спасибо, ты настоящий друг.", gratitude love
|
125 |
+
"Мне так больно из-за потери.", grief sadness
|
126 |
+
"Я счастлив просто быть рядом с тобой.", joy love
|
127 |
+
"Ты моя опора и вдохновение.", love admiration pride
|
128 |
+
"Я нервничаю перед этим испытанием.", nervousness fear
|
129 |
+
"У нас есть все шансы на успех!", optimism joy
|
130 |
+
"Я горжусь, что смог завершить этот проект.", pride joy
|
131 |
+
"Теперь я понял, почему все произошло именно так.", realization curiosity
|
132 |
+
"Облегчение — это самое подходящее слово для описания.", relief gratitude
|
133 |
+
"Прости меня, я очень сожалею о своих словах.", remorse sadness
|
134 |
+
"Мне так грустно, что мы больше не вместе.", sadness grief
|
135 |
+
"Такого я точно не ожидал!", surprise shock
|
136 |
+
"Все спокойно, ничего нового.", neutral
|
137 |
+
"Ты всегда знаешь, как вдохновить.", admiration joy
|
138 |
+
"Это было так смешно, что у меня слезы на глазах.", amusement joy
|
139 |
+
"Ты причинил мне огромную боль, я не могу это простить.", anger grief
|
140 |
+
"Почему я должен постоянно это терпеть?", annoyance frustration
|
141 |
+
"Твоя забота делает мой день лучше.", caring gratitude
|
142 |
+
"Я не понимаю, зачем это вообще нужно.", confusion disapproval
|
143 |
+
"Мне интересно, что скрывается за этой дверью.", curiosity surprise
|
144 |
+
"Я мечтаю об этой машине с самого детства.", desire nostalgia
|
145 |
+
"Как же больно осознавать, что я подвел их.", disappointment remorse
|
146 |
+
"Это было недопустимо и обидно.", disapproval anger
|
147 |
+
"Этот запах вызывает у меня отвращение.", disgust annoyance
|
148 |
+
"Та ситуация была слишком неловкой для всех нас.", embarrassment regret
|
149 |
+
"Я просто не могу дождаться, когда это начнется!", excitement anticipation
|
150 |
+
"Я боюсь, что не успею вовремя.", fear nervousness
|
151 |
+
"Спасибо за все, что ты для меня делаешь.", gratitude love
|
152 |
+
"Моя потеря невыносима.", grief despair
|
153 |
+
"Я так рад, что ты со мной.", joy love
|
154 |
+
"Ты единственный, кто делает меня счастливым.", love
|
155 |
+
"У меня дрожь по телу, я так волнуюсь.", nervousness fear
|
156 |
+
"Все будет замечательно, мы справимся.", optimism pride
|
157 |
+
"Я горжусь тем, что стал частью этого.", pride
|
158 |
+
"Теперь я понял, как это исправить.", realization pride
|
159 |
+
"Слава Богу, это закончилось.", relief
|
160 |
+
"Я сильно жалею о том, что сделал.", remorse sadness
|
161 |
+
"Мне так грустно из-за всего этого.", sadness grief
|
162 |
+
"Этого я точно не ожидал — какая неожиданность!", surprise excitement
|
163 |
+
"Ничего особенного, просто обычный разговор.", neutral
|
164 |
+
"Этот фильм такой смешной, я смеялся весь вечер!", amusement joy
|
165 |
+
"Это видео просто уморительное, его хочется смотреть снова и снова.", amusement curiosity excitement
|
166 |
+
"Почему эти ошибки продолжаются? Это так раздражает!", annoyance anger disappointment
|
167 |
+
"Меня раздражает, что приходится объяснять одно и то же снова и снова.", annoyance frustration disapproval
|
168 |
+
"Ты сделал замечательную работу, продолжай в том же духе.", approval admiration pride
|
169 |
+
"Твой подход заслуживает не только уважения, но и признания.", approval admiration
|
170 |
+
"Я всегда готов поддержать тебя, если понадобится помощь.", caring gratitude love
|
171 |
+
"Спасибо, что заботишься обо мне и всегда рядом.", caring gratitude approval love
|
172 |
+
"Я не понимаю, почему всё так сложно устроено.", confusion frustration
|
173 |
+
"Почему это так сложно? Я ничего не понимаю.", confusion disappointment annoyance
|
174 |
+
"Интересно, почему ты решил сделать именно так?", curiosity realization
|
175 |
+
"Мне очень любопытно, как это всё работает.", curiosity excitement confusion
|
176 |
+
"Я так хочу попробовать этот новый десерт!", desire joy curiosity
|
177 |
+
"Как же хочется отправиться в путешествие этим летом.", desire optimism excitement
|
178 |
+
"Этот запах вызывает у меня отвращение, я даже не могу дышать.", disgust anger annoyance
|
179 |
+
"Поведение этого человека вызывает искреннее отвращение.", disgust disapproval annoyance
|
180 |
+
"Неожиданно увидеть тебя здесь, это так удивительно!", surprise joy
|
181 |
+
"Как неожиданно! Никогда бы не подумал, что это возможно.", surprise realization excitement
|
182 |
+
"Всё прошло совершенно обычно, без сюрпризов.", neutral
|
183 |
+
"Сегодняшний день ничем не отличался от остальных.", neutral disappointment
|
184 |
+
"Этот комикс не только забавный, но и очень умный.", amusement curiosity approval
|
185 |
+
"Меня просто бесит, что ты не слушаешь!", annoyance anger disapproval
|
186 |
+
"Я искренне рад за тебя, ты отлично справился.", approval pride joy
|
187 |
+
"Спасибо за твою заботу, это так важно для меня.", caring gratitude love
|
188 |
+
"Мне сложно понять, что ты имеешь в виду.", confusion frustration curiosity
|
189 |
+
"Мне бы хотелось узнать больше о том, как ты это сделал.", curiosity admiration desire
|
190 |
+
"Я мечтаю о времени, когда смогу путешествовать без ограничений.", desire optimism pride
|
191 |
+
"Эта ситуация была настолько неловкой, что мне до сих пор стыдно.", embarrassment regret remorse
|
192 |
+
"Я рад, что всё закончилось именно так.", relief gratitude joy
|
193 |
+
"Это решение вызывает у меня смешанные чувства: радость и тревогу.", joy fear nervousness
|
194 |
+
"Ничего особенного, просто ещё один день.", neutral
|
195 |
+
"Ты великолепно справился с этой задачей, мне это нравится.", approval admiration
|
196 |
+
"Этот подход выглядит разумным и эффективным.", approval realization
|
197 |
+
"Молодец, ты действительно нашёл правильное решение.", approval pride
|
198 |
+
"Я поддерживаю твою идею, она звучит убедительно.", approval curiosity
|
199 |
+
"Хорошо, что ты подошёл к этому вопросу так серьёзно.", approval caring pride
|
200 |
+
"Этот запах настолько отвратителен, что мне стало плохо.", disgust annoyance
|
201 |
+
"Его действия вызывают у меня только отвращение.", disgust disapproval anger
|
202 |
+
"Эта еда выглядит отвратительно, я не могу её попробовать.", disgust disappointment
|
203 |
+
"Ситуация настолько неприятна, что мне трудно находиться здесь.", disgust nervousness
|
204 |
+
"Его слова звучали так мерзко, что я не мог их выносить.", disgust anger frustration
|
205 |
+
"Всё идёт как обычно, никаких изменений.", neutral
|
206 |
+
"Ничего интересного не произошло, всё осталось по-прежнему.", neutral disappointment
|
207 |
+
"Просто обычный день, без особых событий.", neutral
|
208 |
+
"Всё стабильно, ничего нового.", neutral
|
209 |
+
"Вчерашняя встреча прошла довольно обыденно.", neutral
|
210 |
+
"Я не понимаю, почему этот процесс такой сложный.", confusion frustration
|
211 |
+
"Мне сложно разобраться в этой инструкции.", confusion disapproval
|
212 |
+
"Я не понимаю, зачем они так поступают.", confusion disappointment
|
213 |
+
"Мне интересно, как это устройство работает.", curiosity
|
214 |
+
"Хотел бы узнать больше об этом явлении.", curiosity desire
|
215 |
+
"Этот эксперимент вызывает у меня много вопросов.", curiosity realization
|
216 |
+
"Почему бы нам не попробовать сделать это иначе?", curiosity optimism
|
217 |
+
"Я хотел бы узнать больше о причинах их выбора.", curiosity caring
|
218 |
+
"Я бы хотел, чтобы у нас был шанс попробовать эту стратегию.", desire optimism
|
219 |
+
"Хочется вернуться к этой задаче, чтобы улучшить результат.", desire pride
|
220 |
+
"Как же я хочу добиться лучшего понимания этой темы.", desire curiosity
|
221 |
+
"Мне неприятно видеть такое поведение, оно недопустимо.", disapproval disgust
|
222 |
+
"Этот результат вызывает у меня серьёзные сомнения.", disapproval disappointment
|
223 |
+
"Я не могу согласиться с этим решением, оно кажется ошибочным.", disapproval frustration
|
224 |
+
"Эта идея совершенно противоречит нашим принципам.", disapproval anger
|
225 |
+
"Не могу поверить, что это решение было принято.", disapproval confusion
|
226 |
+
"Неприятно вспоминать эту ситуацию, она была унизительной.", embarrassment remorse
|
227 |
+
"Эта ошибка была настолько очевидной, что мне стало неловко.", embarrassment disapproval
|
228 |
+
"Я всё ещё чувствую неловкость из-за своего поведения.", embarrassment regret
|
229 |
+
"Приятно видеть, как ты стараешься в работе.", approval caring
|
230 |
+
"Хорошо, что ты сделал первый шаг, это важно.", approval pride
|
231 |
+
"Этот документ выглядит аккуратно и профессионально.", approval
|
232 |
+
"Мне нравится, как ты организовал процесс.", approval realization
|
233 |
+
"Отличная работа, продолжай в том же духе.", approval optimism
|
234 |
+
"Этот шум действительно раздражает, невозможно сосредоточиться.", annoyance frustration
|
235 |
+
"Почему ты не слушаешь? Это уже надоело.", annoyance disappointment
|
236 |
+
"Кажется, что я постоянно сталкиваюсь с этими мелкими проблемами.", annoyance disapproval
|
Scripts/EmotionsTest.py
ADDED
@@ -0,0 +1,71 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import pandas as pd
|
2 |
+
import torch
|
3 |
+
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
4 |
+
|
5 |
+
def evaluate_emotions_from_csv(
|
6 |
+
csv_path,
|
7 |
+
model_path,
|
8 |
+
output_csv_path,
|
9 |
+
emotion_columns,
|
10 |
+
device="cuda" if torch.cuda.is_available() else "cpu",
|
11 |
+
threshold=0.1
|
12 |
+
):
|
13 |
+
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
14 |
+
model = AutoModelForSequenceClassification.from_pretrained(model_path)
|
15 |
+
model.to(device)
|
16 |
+
model.eval()
|
17 |
+
|
18 |
+
data = pd.read_csv(csv_path)
|
19 |
+
data['emotions'] = data['emotions'].apply(lambda x: x.split(' '))
|
20 |
+
|
21 |
+
true_emotions_list = []
|
22 |
+
predicted_emotions_list = []
|
23 |
+
correctly_identified = []
|
24 |
+
incorrectly_identified = []
|
25 |
+
undefined = []
|
26 |
+
|
27 |
+
for idx, row in data.iterrows():
|
28 |
+
text = row['text']
|
29 |
+
true_emotions = set(row['emotions'])
|
30 |
+
|
31 |
+
inputs = tokenizer(text, return_tensors="pt", padding="max_length", truncation=True, max_length=128).to(device)
|
32 |
+
with torch.no_grad():
|
33 |
+
logits = model(**inputs).logits
|
34 |
+
probabilities = torch.sigmoid(logits).squeeze().cpu().numpy()
|
35 |
+
|
36 |
+
predicted_emotions = set(
|
37 |
+
emotion for emotion, prob in zip(emotion_columns, probabilities) if prob > threshold
|
38 |
+
)
|
39 |
+
|
40 |
+
correct = true_emotions & predicted_emotions
|
41 |
+
incorrect = predicted_emotions - true_emotions
|
42 |
+
undefined_emotions = true_emotions - predicted_emotions
|
43 |
+
|
44 |
+
true_emotions_list.append(' '.join(true_emotions))
|
45 |
+
predicted_emotions_list.append(' '.join(predicted_emotions))
|
46 |
+
correctly_identified.append(' '.join(correct) if correct else 'None')
|
47 |
+
incorrectly_identified.append(' '.join(incorrect) if incorrect else 'None')
|
48 |
+
undefined.append(' '.join(undefined_emotions) if undefined_emotions else 'None')
|
49 |
+
|
50 |
+
results_df = pd.DataFrame({
|
51 |
+
"text": data['text'],
|
52 |
+
"true emotions": true_emotions_list,
|
53 |
+
"predict emotions": predicted_emotions_list,
|
54 |
+
"correctly Identified": correctly_identified,
|
55 |
+
"incorrectly Identified": incorrectly_identified,
|
56 |
+
"undefined": undefined,
|
57 |
+
})
|
58 |
+
|
59 |
+
results_df.to_csv(output_csv_path, index=False)
|
60 |
+
print(f"Результаты сохранены в {output_csv_path}")
|
61 |
+
|
62 |
+
csv_path = "RuBert-tiny2-EmotionsDetected/Dstasets/testEmotionDetected.csv"
|
63 |
+
model_path = "RuBert-tiny2-EmotionsDetected"
|
64 |
+
output_csv_path = "RuBert-tiny2-EmotionsDetected/Dstasets/Emotions_detected.csv"
|
65 |
+
emotion_columns = [
|
66 |
+
"admiration", "amusement", "anger", "annoyance", "approval", "caring", "confusion", "curiosity", "desire",
|
67 |
+
"disappointment", "disapproval", "disgust", "embarrassment", "excitement", "fear", "gratitude", "grief", "joy",
|
68 |
+
"love", "nervousness", "optimism", "pride", "realization", "relief", "remorse", "sadness", "surprise", "neutral"
|
69 |
+
]
|
70 |
+
|
71 |
+
evaluate_emotions_from_csv(csv_path, model_path, output_csv_path, emotion_columns, threshold=0.2)
|
Scripts/RuBert-tiny2-fine-tune-EmotionsDetected/emotiondetected.ipynb
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.10.12","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"gpu","dataSources":[{"sourceId":10605153,"sourceType":"datasetVersion","datasetId":6564827}],"dockerImageVersionId":30840,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":true}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"import warnings\nimport os\n\nos.environ[\"WANDB_DISABLED\"] = \"true\"\nwarnings.filterwarnings(\"ignore\")","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true,"execution":{"iopub.status.busy":"2025-01-29T00:38:58.678784Z","iopub.execute_input":"2025-01-29T00:38:58.679128Z","iopub.status.idle":"2025-01-29T00:38:58.683044Z","shell.execute_reply.started":"2025-01-29T00:38:58.679099Z","shell.execute_reply":"2025-01-29T00:38:58.682291Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"import pandas as pd\nimport numpy as np\nimport torch\nfrom sklearn.model_selection import train_test_split\nfrom sklearn.metrics import f1_score\nfrom transformers import AutoTokenizer, AutoModelForSequenceClassification, TrainingArguments, Trainer\nfrom torch.utils.data import Dataset","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T00:39:01.650272Z","iopub.execute_input":"2025-01-29T00:39:01.650566Z","iopub.status.idle":"2025-01-29T00:39:01.654590Z","shell.execute_reply.started":"2025-01-29T00:39:01.650544Z","shell.execute_reply":"2025-01-29T00:39:01.653777Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"data = pd.read_csv('/kaggle/input/ru-go-emotions-raw/ru-go-emotions-raw.csv')\nemotion_columns = data.columns[10:]\ndata = data.dropna(subset=['ru_text']) ","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T00:39:04.795445Z","iopub.execute_input":"2025-01-29T00:39:04.795746Z","iopub.status.idle":"2025-01-29T00:39:06.458939Z","shell.execute_reply.started":"2025-01-29T00:39:04.795722Z","shell.execute_reply":"2025-01-29T00:39:06.458241Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"train_texts, val_texts, train_labels, val_labels = train_test_split(\n data['ru_text'].tolist(),\n data[emotion_columns].values,\n test_size=0.1,\n random_state=42\n)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T00:39:07.803815Z","iopub.execute_input":"2025-01-29T00:39:07.804102Z","iopub.status.idle":"2025-01-29T00:39:07.976716Z","shell.execute_reply.started":"2025-01-29T00:39:07.804080Z","shell.execute_reply":"2025-01-29T00:39:07.975808Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"tokenizer = AutoTokenizer.from_pretrained(\"cointegrated/rubert-tiny2\")\n\ndef tokenize_function(texts):\n return tokenizer(texts, padding=\"max_length\", truncation=True, max_length=128)\n\ntrain_encodings = tokenize_function(train_texts)\nval_encodings = tokenize_function(val_texts)\n","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T00:39:10.715401Z","iopub.execute_input":"2025-01-29T00:39:10.715712Z","iopub.status.idle":"2025-01-29T00:39:25.842366Z","shell.execute_reply.started":"2025-01-29T00:39:10.715688Z","shell.execute_reply":"2025-01-29T00:39:25.841676Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"class EmotionDataset(Dataset):\n def __init__(self, encodings, labels):\n self.encodings = encodings\n self.labels = labels\n\n def __len__(self):\n return len(self.labels)\n\n def __getitem__(self, idx):\n item = {key: torch.tensor(val[idx]) for key, val in self.encodings.items()}\n item['labels'] = torch.tensor(self.labels[idx]).float()\n return item\n\ntrain_dataset = EmotionDataset(train_encodings, train_labels)\nval_dataset = EmotionDataset(val_encodings, val_labels)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T00:39:25.843495Z","iopub.execute_input":"2025-01-29T00:39:25.843806Z","iopub.status.idle":"2025-01-29T00:39:25.849286Z","shell.execute_reply.started":"2025-01-29T00:39:25.843776Z","shell.execute_reply":"2025-01-29T00:39:25.848623Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"model = AutoModelForSequenceClassification.from_pretrained(\"cointegrated/rubert-tiny2\", num_labels=len(emotion_columns))","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T00:39:25.850691Z","iopub.execute_input":"2025-01-29T00:39:25.850970Z","iopub.status.idle":"2025-01-29T00:39:27.423163Z","shell.execute_reply.started":"2025-01-29T00:39:25.850949Z","shell.execute_reply":"2025-01-29T00:39:27.422523Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"training_args = TrainingArguments(\n output_dir=\"./results\",\n evaluation_strategy=\"epoch\",\n save_strategy=\"epoch\",\n logging_dir=\"./logs\",\n logging_steps=100,\n per_device_train_batch_size=16,\n per_device_eval_batch_size=16,\n num_train_epochs=1,\n weight_decay=0.01,\n learning_rate=1e-5,\n save_total_limit=2,\n load_best_model_at_end=True,\n metric_for_best_model=\"f1\",\n greater_is_better=True,\n)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T03:22:49.776926Z","iopub.execute_input":"2025-01-29T03:22:49.777272Z","iopub.status.idle":"2025-01-29T03:22:49.807148Z","shell.execute_reply.started":"2025-01-29T03:22:49.777216Z","shell.execute_reply":"2025-01-29T03:22:49.806312Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"def compute_metrics(eval_pred):\n logits, labels = eval_pred\n preds = (torch.sigmoid(torch.tensor(logits)) > 0.5).int().numpy()\n f1 = f1_score(labels, preds, average=\"weighted\")\n return {\"f1\": f1}","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T00:40:12.373628Z","iopub.execute_input":"2025-01-29T00:40:12.373925Z","iopub.status.idle":"2025-01-29T00:40:12.378434Z","shell.execute_reply.started":"2025-01-29T00:40:12.373903Z","shell.execute_reply":"2025-01-29T00:40:12.377447Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"trainer = Trainer(\n model=model,\n args=training_args,\n train_dataset=train_dataset,\n eval_dataset=val_dataset,\n tokenizer=tokenizer,\n compute_metrics=compute_metrics,\n)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T03:22:57.004354Z","iopub.execute_input":"2025-01-29T03:22:57.004645Z","iopub.status.idle":"2025-01-29T03:22:57.015351Z","shell.execute_reply.started":"2025-01-29T03:22:57.004620Z","shell.execute_reply":"2025-01-29T03:22:57.014668Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"trainer.train()","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T03:22:58.692598Z","iopub.execute_input":"2025-01-29T03:22:58.692886Z","iopub.status.idle":"2025-01-29T03:27:33.373784Z","shell.execute_reply.started":"2025-01-29T03:22:58.692863Z","shell.execute_reply":"2025-01-29T03:27:33.373175Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"model.save_pretrained(\"./emotion_model\")\ntokenizer.save_pretrained(\"./emotion_model\")","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T03:17:36.721898Z","iopub.execute_input":"2025-01-29T03:17:36.722196Z","iopub.status.idle":"2025-01-29T03:17:37.192040Z","shell.execute_reply.started":"2025-01-29T03:17:36.722174Z","shell.execute_reply":"2025-01-29T03:17:37.191318Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"def predict_emotions(\n text, \n model, \n tokenizer, \n emotion_columns, \n device=\"cuda\" if torch.cuda.is_available() else \"cpu\", \n threshold=0.1\n):\n\n emotion_translations = {\n \"admiration\": \"восхищение\",\n \"amusement\": \"веселье\",\n \"anger\": \"злость\",\n \"annoyance\": \"раздражение\",\n \"approval\": \"одобрение\",\n \"caring\": \"забота\",\n \"confusion\": \"непонимание\",\n \"curiosity\": \"любопытство\",\n \"desire\": \"желание\",\n \"disappointment\": \"разочарование\",\n \"disapproval\": \"неодобрение\",\n \"disgust\": \"отвращение\",\n \"embarrassment\": \"смущение\",\n \"excitement\": \"возбуждение\",\n \"fear\": \"страх\",\n \"gratitude\": \"признательность\",\n \"grief\": \"горе\",\n \"joy\": \"радость\",\n \"love\": \"любовь\",\n \"nervousness\": \"нервозность\",\n \"optimism\": \"оптимизм\",\n \"pride\": \"гордость\",\n \"realization\": \"осознание\",\n \"relief\": \"облегчение\",\n \"remorse\": \"раскаяние\",\n \"sadness\": \"грусть\",\n \"surprise\": \"удивление\",\n \"neutral\": \"нейтральность\",\n }\n\n model.to(device)\n model.eval()\n inputs = tokenizer(text, return_tensors=\"pt\", padding=\"max_length\", truncation=True, max_length=128).to(device)\n with torch.no_grad():\n logits = model(**inputs).logits\n probabilities = torch.sigmoid(logits).squeeze().cpu().numpy()\n\n predictions = {\n f\"{emotion} ({emotion_translations[emotion]})\": prob\n for emotion, prob in zip(emotion_columns, probabilities) if prob > threshold\n }\n\n sorted_predictions = dict(sorted(predictions.items(), key=lambda item: item[1], reverse=True))\n \n return sorted_predictions\n","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T03:17:48.694048Z","iopub.execute_input":"2025-01-29T03:17:48.694355Z","iopub.status.idle":"2025-01-29T03:17:48.701764Z","shell.execute_reply.started":"2025-01-29T03:17:48.694331Z","shell.execute_reply":"2025-01-29T03:17:48.700844Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"example_text = \"Как же я рад!\"\npredictions = predict_emotions(example_text, model, tokenizer, emotion_columns)\nprint(\"Emotions:\", predictions)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T03:28:41.045221Z","iopub.execute_input":"2025-01-29T03:28:41.045575Z","iopub.status.idle":"2025-01-29T03:28:41.057190Z","shell.execute_reply.started":"2025-01-29T03:28:41.045547Z","shell.execute_reply":"2025-01-29T03:28:41.056478Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"example_text = \"Как же я не рад!\"\npredictions = predict_emotions(example_text, model, tokenizer, emotion_columns)\nprint(\"Emotions:\", predictions)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T03:28:49.329257Z","iopub.execute_input":"2025-01-29T03:28:49.329561Z","iopub.status.idle":"2025-01-29T03:28:49.341461Z","shell.execute_reply.started":"2025-01-29T03:28:49.329537Z","shell.execute_reply":"2025-01-29T03:28:49.340540Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"example_text = \"Всё очень плохо!\"\npredictions = predict_emotions(example_text, model, tokenizer, emotion_columns)\nprint(\"Emotions:\", predictions)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T03:28:59.880762Z","iopub.execute_input":"2025-01-29T03:28:59.881090Z","iopub.status.idle":"2025-01-29T03:28:59.892709Z","shell.execute_reply.started":"2025-01-29T03:28:59.881059Z","shell.execute_reply":"2025-01-29T03:28:59.891839Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"example_text = \"ого! вот это да!\"\npredictions = predict_emotions(example_text, model, tokenizer, emotion_columns)\nprint(\"Emotions:\", predictions)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T03:18:32.552701Z","iopub.execute_input":"2025-01-29T03:18:32.553029Z","iopub.status.idle":"2025-01-29T03:18:32.563438Z","shell.execute_reply.started":"2025-01-29T03:18:32.553001Z","shell.execute_reply":"2025-01-29T03:18:32.562734Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"trainer.save_model()\n","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T03:18:41.909667Z","iopub.execute_input":"2025-01-29T03:18:41.909954Z","iopub.status.idle":"2025-01-29T03:18:42.391982Z","shell.execute_reply.started":"2025-01-29T03:18:41.909932Z","shell.execute_reply":"2025-01-29T03:18:42.391053Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"import os\nimport subprocess\nfrom IPython.display import FileLink, display\n\ndef download_file(path, download_file_name):\n os.chdir('/kaggle/working/')\n zip_name = f\"/kaggle/working/{download_file_name}.zip\"\n command = f\"zip {zip_name} {path} -r\"\n result = subprocess.run(command, shell=True, capture_output=True, text=True)\n if result.returncode != 0:\n print(\"Unable to run zip command!\")\n print(result.stderr)\n return\n display(FileLink(f'{download_file_name}.zip'))","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T02:29:26.763441Z","iopub.execute_input":"2025-01-29T02:29:26.763737Z","iopub.status.idle":"2025-01-29T02:29:26.768654Z","shell.execute_reply.started":"2025-01-29T02:29:26.763713Z","shell.execute_reply":"2025-01-29T02:29:26.767838Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"download_file('/kaggle/working/emotion_model', 'Emotions_model-04')","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-01-29T03:18:52.390465Z","iopub.execute_input":"2025-01-29T03:18:52.390787Z","iopub.status.idle":"2025-01-29T03:18:58.791398Z","shell.execute_reply.started":"2025-01-29T03:18:52.390758Z","shell.execute_reply":"2025-01-29T03:18:58.790673Z"}},"outputs":[],"execution_count":null}]}
|
Scripts/TermoMap.py
ADDED
@@ -0,0 +1,111 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import seaborn as sns
|
2 |
+
import matplotlib.pyplot as plt
|
3 |
+
import pandas as pd
|
4 |
+
import numpy as np
|
5 |
+
|
6 |
+
def plot_emotion_confusion_matrix(results_df, emotion_columns):
|
7 |
+
|
8 |
+
correct_count = {emotion: 0 for emotion in emotion_columns}
|
9 |
+
incorrect_count = {emotion: 0 for emotion in emotion_columns}
|
10 |
+
undefined_count = {emotion: 0 for emotion in emotion_columns}
|
11 |
+
|
12 |
+
for idx, row in results_df.iterrows():
|
13 |
+
true_emotions = set(row['true emotions'].split()) if isinstance(row['true emotions'], str) else set()
|
14 |
+
predicted_emotions = set(row['predict emotions'].split()) if isinstance(row['predict emotions'], str) else set()
|
15 |
+
|
16 |
+
for emotion in emotion_columns:
|
17 |
+
if emotion in true_emotions and emotion in predicted_emotions:
|
18 |
+
correct_count[emotion] += 1
|
19 |
+
elif emotion in predicted_emotions and emotion not in true_emotions:
|
20 |
+
incorrect_count[emotion] += 1
|
21 |
+
elif emotion in true_emotions and emotion not in predicted_emotions:
|
22 |
+
undefined_count[emotion] += 1
|
23 |
+
|
24 |
+
data = []
|
25 |
+
for emotion in emotion_columns:
|
26 |
+
data.append([
|
27 |
+
correct_count[emotion],
|
28 |
+
incorrect_count[emotion],
|
29 |
+
undefined_count[emotion]
|
30 |
+
])
|
31 |
+
|
32 |
+
heatmap_df = pd.DataFrame(data, columns=["Correctly Identified", "Incorrectly Identified", "Undefined"], index=emotion_columns)
|
33 |
+
|
34 |
+
num_examples = len(results_df)
|
35 |
+
|
36 |
+
plt.figure(figsize=(10, 12))
|
37 |
+
sns.heatmap(heatmap_df, annot=True, cmap="Blues", fmt="d", cbar=False)
|
38 |
+
plt.title(f"Emotion Prediction Confusion Matrix (Examples: {num_examples})")
|
39 |
+
plt.xlabel("Prediction Status")
|
40 |
+
plt.ylabel("Emotion")
|
41 |
+
plt.xticks(rotation=45, ha="right")
|
42 |
+
plt.tight_layout()
|
43 |
+
plt.show()
|
44 |
+
|
45 |
+
|
46 |
+
def plot_true_emotion_frequency(results_df, emotion_columns):
|
47 |
+
|
48 |
+
true_emotion_count = {emotion: 0 for emotion in emotion_columns}
|
49 |
+
|
50 |
+
for idx, row in results_df.iterrows():
|
51 |
+
true_emotions = set(row['true emotions'].split()) if isinstance(row['true emotions'], str) else set()
|
52 |
+
|
53 |
+
for emotion in emotion_columns:
|
54 |
+
if emotion in true_emotions:
|
55 |
+
true_emotion_count[emotion] += 1
|
56 |
+
|
57 |
+
data = []
|
58 |
+
for emotion in emotion_columns:
|
59 |
+
data.append([true_emotion_count[emotion]])
|
60 |
+
|
61 |
+
heatmap_df = pd.DataFrame(data, columns=["True Emotion Count"], index=emotion_columns)
|
62 |
+
|
63 |
+
plt.figure(figsize=(10, 12))
|
64 |
+
sns.heatmap(heatmap_df, annot=True, cmap="YlGnBu", fmt="d", cbar=False)
|
65 |
+
plt.title(f"True Emotion Frequency (Examples: {len(results_df)})")
|
66 |
+
plt.xlabel("True Emotion Count")
|
67 |
+
plt.ylabel("Emotion")
|
68 |
+
plt.xticks(rotation=45, ha="right")
|
69 |
+
plt.tight_layout()
|
70 |
+
plt.show()
|
71 |
+
|
72 |
+
|
73 |
+
def plot_predicted_emotion_frequency(results_df, emotion_columns):
|
74 |
+
|
75 |
+
predicted_emotion_count = {emotion: 0 for emotion in emotion_columns}
|
76 |
+
|
77 |
+
for idx, row in results_df.iterrows():
|
78 |
+
predicted_emotions = set(row['predict emotions'].split()) if isinstance(row['predict emotions'], str) else set()
|
79 |
+
|
80 |
+
for emotion in emotion_columns:
|
81 |
+
if emotion in predicted_emotions:
|
82 |
+
predicted_emotion_count[emotion] += 1
|
83 |
+
|
84 |
+
data = []
|
85 |
+
for emotion in emotion_columns:
|
86 |
+
data.append([predicted_emotion_count[emotion]])
|
87 |
+
|
88 |
+
heatmap_df = pd.DataFrame(data, columns=["Predicted Emotion Count"], index=emotion_columns)
|
89 |
+
|
90 |
+
plt.figure(figsize=(10, 12))
|
91 |
+
sns.heatmap(heatmap_df, annot=True, cmap="YlOrRd", fmt="d", cbar=False)
|
92 |
+
plt.title(f"Predicted Emotion Frequency (Examples: {len(results_df)})")
|
93 |
+
plt.xlabel("Predicted Emotion Count")
|
94 |
+
plt.ylabel("Emotion")
|
95 |
+
plt.xticks(rotation=45, ha="right")
|
96 |
+
plt.tight_layout()
|
97 |
+
plt.show()
|
98 |
+
|
99 |
+
|
100 |
+
csv_path = "RuBert-tiny2-EmotionsDetected/Dstasets/Emotions_detected.csv"
|
101 |
+
results_df = pd.read_csv(csv_path)
|
102 |
+
|
103 |
+
emotion_columns = [
|
104 |
+
"admiration", "amusement", "anger", "annoyance", "approval", "caring", "confusion", "curiosity", "desire",
|
105 |
+
"disappointment", "disapproval", "disgust", "embarrassment", "excitement", "fear", "gratitude", "grief", "joy",
|
106 |
+
"love", "nervousness", "optimism", "pride", "realization", "relief", "remorse", "sadness", "surprise", "neutral"
|
107 |
+
]
|
108 |
+
|
109 |
+
plot_true_emotion_frequency(results_df, emotion_columns)
|
110 |
+
plot_predicted_emotion_frequency(results_df, emotion_columns)
|
111 |
+
plot_emotion_confusion_matrix(results_df, emotion_columns)
|
Versions/Emotions_model-01.zip
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1064d4a6d90d50d6205b587f2f4ec21adc29d8a86fa4388aa0d04361bb30f2bb
|
3 |
+
size 108521000
|
Versions/Emotions_model-02.zip
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:46b2d934fd4c3fac69ea0ea608a8a2e38a480950f282fafeef210bd8bfa52dee
|
3 |
+
size 108519899
|
Versions/Emotions_model-03.zip
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:65f09107eb1fd1d6d750745d91379ae73a4d0be388061bb9df5dac6fb8bb24de
|
3 |
+
size 108517162
|
Versions/Emotions_model-04.zip
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c08ddf05bc5079e3c4361bf59eafe0f6c2c371cb46c821f5b2c5a4e75e645539
|
3 |
+
size 108513058
|
config.json
ADDED
@@ -0,0 +1,88 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "cointegrated/rubert-tiny2",
|
3 |
+
"architectures": [
|
4 |
+
"BertForSequenceClassification"
|
5 |
+
],
|
6 |
+
"attention_probs_dropout_prob": 0.1,
|
7 |
+
"classifier_dropout": null,
|
8 |
+
"emb_size": 312,
|
9 |
+
"gradient_checkpointing": false,
|
10 |
+
"hidden_act": "gelu",
|
11 |
+
"hidden_dropout_prob": 0.1,
|
12 |
+
"hidden_size": 312,
|
13 |
+
"id2label": {
|
14 |
+
"0": "LABEL_0",
|
15 |
+
"1": "LABEL_1",
|
16 |
+
"2": "LABEL_2",
|
17 |
+
"3": "LABEL_3",
|
18 |
+
"4": "LABEL_4",
|
19 |
+
"5": "LABEL_5",
|
20 |
+
"6": "LABEL_6",
|
21 |
+
"7": "LABEL_7",
|
22 |
+
"8": "LABEL_8",
|
23 |
+
"9": "LABEL_9",
|
24 |
+
"10": "LABEL_10",
|
25 |
+
"11": "LABEL_11",
|
26 |
+
"12": "LABEL_12",
|
27 |
+
"13": "LABEL_13",
|
28 |
+
"14": "LABEL_14",
|
29 |
+
"15": "LABEL_15",
|
30 |
+
"16": "LABEL_16",
|
31 |
+
"17": "LABEL_17",
|
32 |
+
"18": "LABEL_18",
|
33 |
+
"19": "LABEL_19",
|
34 |
+
"20": "LABEL_20",
|
35 |
+
"21": "LABEL_21",
|
36 |
+
"22": "LABEL_22",
|
37 |
+
"23": "LABEL_23",
|
38 |
+
"24": "LABEL_24",
|
39 |
+
"25": "LABEL_25",
|
40 |
+
"26": "LABEL_26",
|
41 |
+
"27": "LABEL_27"
|
42 |
+
},
|
43 |
+
"initializer_range": 0.02,
|
44 |
+
"intermediate_size": 600,
|
45 |
+
"label2id": {
|
46 |
+
"LABEL_0": 0,
|
47 |
+
"LABEL_1": 1,
|
48 |
+
"LABEL_10": 10,
|
49 |
+
"LABEL_11": 11,
|
50 |
+
"LABEL_12": 12,
|
51 |
+
"LABEL_13": 13,
|
52 |
+
"LABEL_14": 14,
|
53 |
+
"LABEL_15": 15,
|
54 |
+
"LABEL_16": 16,
|
55 |
+
"LABEL_17": 17,
|
56 |
+
"LABEL_18": 18,
|
57 |
+
"LABEL_19": 19,
|
58 |
+
"LABEL_2": 2,
|
59 |
+
"LABEL_20": 20,
|
60 |
+
"LABEL_21": 21,
|
61 |
+
"LABEL_22": 22,
|
62 |
+
"LABEL_23": 23,
|
63 |
+
"LABEL_24": 24,
|
64 |
+
"LABEL_25": 25,
|
65 |
+
"LABEL_26": 26,
|
66 |
+
"LABEL_27": 27,
|
67 |
+
"LABEL_3": 3,
|
68 |
+
"LABEL_4": 4,
|
69 |
+
"LABEL_5": 5,
|
70 |
+
"LABEL_6": 6,
|
71 |
+
"LABEL_7": 7,
|
72 |
+
"LABEL_8": 8,
|
73 |
+
"LABEL_9": 9
|
74 |
+
},
|
75 |
+
"layer_norm_eps": 1e-12,
|
76 |
+
"max_position_embeddings": 2048,
|
77 |
+
"model_type": "bert",
|
78 |
+
"num_attention_heads": 12,
|
79 |
+
"num_hidden_layers": 3,
|
80 |
+
"pad_token_id": 0,
|
81 |
+
"position_embedding_type": "absolute",
|
82 |
+
"problem_type": "multi_label_classification",
|
83 |
+
"torch_dtype": "float32",
|
84 |
+
"transformers_version": "4.47.0",
|
85 |
+
"type_vocab_size": 2,
|
86 |
+
"use_cache": true,
|
87 |
+
"vocab_size": 83828
|
88 |
+
}
|
img/Comparison.png
ADDED
![]() |
img/PredictedEmotion.png
ADDED
![]() |
img/TrueEmotion.png
ADDED
![]() |
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9308c79318a13ad00cc5bfb031256b161e2b9a9a9e33ff20bdf2bcfa5ca5aeed
|
3 |
+
size 116816688
|
special_tokens_map.json
ADDED
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"cls_token": {
|
3 |
+
"content": "[CLS]",
|
4 |
+
"lstrip": false,
|
5 |
+
"normalized": false,
|
6 |
+
"rstrip": false,
|
7 |
+
"single_word": false
|
8 |
+
},
|
9 |
+
"mask_token": {
|
10 |
+
"content": "[MASK]",
|
11 |
+
"lstrip": false,
|
12 |
+
"normalized": false,
|
13 |
+
"rstrip": false,
|
14 |
+
"single_word": false
|
15 |
+
},
|
16 |
+
"pad_token": {
|
17 |
+
"content": "[PAD]",
|
18 |
+
"lstrip": false,
|
19 |
+
"normalized": false,
|
20 |
+
"rstrip": false,
|
21 |
+
"single_word": false
|
22 |
+
},
|
23 |
+
"sep_token": {
|
24 |
+
"content": "[SEP]",
|
25 |
+
"lstrip": false,
|
26 |
+
"normalized": false,
|
27 |
+
"rstrip": false,
|
28 |
+
"single_word": false
|
29 |
+
},
|
30 |
+
"unk_token": {
|
31 |
+
"content": "[UNK]",
|
32 |
+
"lstrip": false,
|
33 |
+
"normalized": false,
|
34 |
+
"rstrip": false,
|
35 |
+
"single_word": false
|
36 |
+
}
|
37 |
+
}
|
test.ipynb
ADDED
@@ -0,0 +1,179 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"cells": [
|
3 |
+
{
|
4 |
+
"cell_type": "code",
|
5 |
+
"execution_count": 1,
|
6 |
+
"metadata": {},
|
7 |
+
"outputs": [],
|
8 |
+
"source": [
|
9 |
+
"import pandas as pd\n",
|
10 |
+
"import torch\n",
|
11 |
+
"from transformers import AutoTokenizer, AutoModelForSequenceClassification"
|
12 |
+
]
|
13 |
+
},
|
14 |
+
{
|
15 |
+
"cell_type": "code",
|
16 |
+
"execution_count": 2,
|
17 |
+
"metadata": {},
|
18 |
+
"outputs": [
|
19 |
+
{
|
20 |
+
"name": "stderr",
|
21 |
+
"output_type": "stream",
|
22 |
+
"text": [
|
23 |
+
"c:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\huggingface_hub\\file_download.py:1142: FutureWarning: `resume_download` is deprecated and will be removed in version 1.0.0. Downloads always resume when possible. If you want to force a new download, use `force_download=True`.\n",
|
24 |
+
" warnings.warn(\n"
|
25 |
+
]
|
26 |
+
},
|
27 |
+
{
|
28 |
+
"ename": "OSError",
|
29 |
+
"evalue": "AniMAntZeZo/RuBert-tiny2-EmotionsDetected does not appear to have a file named config.json. Checkout 'https://huggingface.co/AniMAntZeZo/RuBert-tiny2-EmotionsDetected/main' for available files.",
|
30 |
+
"output_type": "error",
|
31 |
+
"traceback": [
|
32 |
+
"\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
|
33 |
+
"\u001b[1;31mHTTPError\u001b[0m Traceback (most recent call last)",
|
34 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\huggingface_hub\\utils\\_http.py:406\u001b[0m, in \u001b[0;36mhf_raise_for_status\u001b[1;34m(response, endpoint_name)\u001b[0m\n\u001b[0;32m 405\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[1;32m--> 406\u001b[0m \u001b[43mresponse\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mraise_for_status\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n\u001b[0;32m 407\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m HTTPError \u001b[38;5;28;01mas\u001b[39;00m e:\n",
|
35 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\requests\\models.py:1024\u001b[0m, in \u001b[0;36mResponse.raise_for_status\u001b[1;34m(self)\u001b[0m\n\u001b[0;32m 1023\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m http_error_msg:\n\u001b[1;32m-> 1024\u001b[0m \u001b[38;5;28;01mraise\u001b[39;00m HTTPError(http_error_msg, response\u001b[38;5;241m=\u001b[39m\u001b[38;5;28mself\u001b[39m)\n",
|
36 |
+
"\u001b[1;31mHTTPError\u001b[0m: 404 Client Error: Not Found for url: https://huggingface.co/AniMAntZeZo/RuBert-tiny2-EmotionsDetected/resolve/main/config.json",
|
37 |
+
"\nThe above exception was the direct cause of the following exception:\n",
|
38 |
+
"\u001b[1;31mEntryNotFoundError\u001b[0m Traceback (most recent call last)",
|
39 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\transformers\\utils\\hub.py:385\u001b[0m, in \u001b[0;36mcached_file\u001b[1;34m(path_or_repo_id, filename, cache_dir, force_download, resume_download, proxies, token, revision, local_files_only, subfolder, repo_type, user_agent, _raise_exceptions_for_missing_entries, _raise_exceptions_for_connection_errors, _commit_hash, **deprecated_kwargs)\u001b[0m\n\u001b[0;32m 383\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m 384\u001b[0m \u001b[38;5;66;03m# Load from URL or cache if already cached\u001b[39;00m\n\u001b[1;32m--> 385\u001b[0m resolved_file \u001b[38;5;241m=\u001b[39m \u001b[43mhf_hub_download\u001b[49m\u001b[43m(\u001b[49m\n\u001b[0;32m 386\u001b[0m \u001b[43m \u001b[49m\u001b[43mpath_or_repo_id\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 387\u001b[0m \u001b[43m \u001b[49m\u001b[43mfilename\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 388\u001b[0m \u001b[43m \u001b[49m\u001b[43msubfolder\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mNone\u001b[39;49;00m\u001b[43m \u001b[49m\u001b[38;5;28;43;01mif\u001b[39;49;00m\u001b[43m \u001b[49m\u001b[38;5;28;43mlen\u001b[39;49m\u001b[43m(\u001b[49m\u001b[43msubfolder\u001b[49m\u001b[43m)\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m==\u001b[39;49m\u001b[43m \u001b[49m\u001b[38;5;241;43m0\u001b[39;49m\u001b[43m \u001b[49m\u001b[38;5;28;43;01melse\u001b[39;49;00m\u001b[43m \u001b[49m\u001b[43msubfolder\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 389\u001b[0m \u001b[43m \u001b[49m\u001b[43mrepo_type\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrepo_type\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 390\u001b[0m \u001b[43m \u001b[49m\u001b[43mrevision\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrevision\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 391\u001b[0m \u001b[43m \u001b[49m\u001b[43mcache_dir\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mcache_dir\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 392\u001b[0m \u001b[43m \u001b[49m\u001b[43muser_agent\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43muser_agent\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 393\u001b[0m \u001b[43m \u001b[49m\u001b[43mforce_download\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mforce_download\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 394\u001b[0m \u001b[43m \u001b[49m\u001b[43mproxies\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mproxies\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 395\u001b[0m \u001b[43m \u001b[49m\u001b[43mresume_download\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mresume_download\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 396\u001b[0m \u001b[43m \u001b[49m\u001b[43mtoken\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtoken\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 397\u001b[0m \u001b[43m \u001b[49m\u001b[43mlocal_files_only\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mlocal_files_only\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 398\u001b[0m \u001b[43m \u001b[49m\u001b[43m)\u001b[49m\n\u001b[0;32m 399\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m GatedRepoError \u001b[38;5;28;01mas\u001b[39;00m e:\n",
|
40 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\huggingface_hub\\utils\\_deprecation.py:101\u001b[0m, in \u001b[0;36m_deprecate_arguments.<locals>._inner_deprecate_positional_args.<locals>.inner_f\u001b[1;34m(*args, **kwargs)\u001b[0m\n\u001b[0;32m 100\u001b[0m warnings\u001b[38;5;241m.\u001b[39mwarn(message, \u001b[38;5;167;01mFutureWarning\u001b[39;00m)\n\u001b[1;32m--> 101\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m f(\u001b[38;5;241m*\u001b[39margs, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs)\n",
|
41 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\huggingface_hub\\utils\\_validators.py:114\u001b[0m, in \u001b[0;36mvalidate_hf_hub_args.<locals>._inner_fn\u001b[1;34m(*args, **kwargs)\u001b[0m\n\u001b[0;32m 112\u001b[0m kwargs \u001b[38;5;241m=\u001b[39m smoothly_deprecate_use_auth_token(fn_name\u001b[38;5;241m=\u001b[39mfn\u001b[38;5;241m.\u001b[39m\u001b[38;5;18m__name__\u001b[39m, has_token\u001b[38;5;241m=\u001b[39mhas_token, kwargs\u001b[38;5;241m=\u001b[39mkwargs)\n\u001b[1;32m--> 114\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m fn(\u001b[38;5;241m*\u001b[39margs, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs)\n",
|
42 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\huggingface_hub\\file_download.py:1232\u001b[0m, in \u001b[0;36mhf_hub_download\u001b[1;34m(repo_id, filename, subfolder, repo_type, revision, library_name, library_version, cache_dir, local_dir, user_agent, force_download, proxies, etag_timeout, token, local_files_only, headers, endpoint, legacy_cache_layout, resume_download, force_filename, local_dir_use_symlinks)\u001b[0m\n\u001b[0;32m 1231\u001b[0m \u001b[38;5;28;01melse\u001b[39;00m:\n\u001b[1;32m-> 1232\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[43m_hf_hub_download_to_cache_dir\u001b[49m\u001b[43m(\u001b[49m\n\u001b[0;32m 1233\u001b[0m \u001b[43m \u001b[49m\u001b[38;5;66;43;03m# Destination\u001b[39;49;00m\n\u001b[0;32m 1234\u001b[0m \u001b[43m \u001b[49m\u001b[43mcache_dir\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mcache_dir\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1235\u001b[0m \u001b[43m \u001b[49m\u001b[38;5;66;43;03m# File info\u001b[39;49;00m\n\u001b[0;32m 1236\u001b[0m \u001b[43m \u001b[49m\u001b[43mrepo_id\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrepo_id\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1237\u001b[0m \u001b[43m \u001b[49m\u001b[43mfilename\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mfilename\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1238\u001b[0m \u001b[43m \u001b[49m\u001b[43mrepo_type\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrepo_type\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1239\u001b[0m \u001b[43m \u001b[49m\u001b[43mrevision\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrevision\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1240\u001b[0m \u001b[43m \u001b[49m\u001b[38;5;66;43;03m# HTTP info\u001b[39;49;00m\n\u001b[0;32m 1241\u001b[0m \u001b[43m \u001b[49m\u001b[43mendpoint\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mendpoint\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1242\u001b[0m \u001b[43m \u001b[49m\u001b[43metag_timeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43metag_timeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1243\u001b[0m \u001b[43m \u001b[49m\u001b[43mheaders\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1244\u001b[0m \u001b[43m \u001b[49m\u001b[43mproxies\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mproxies\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1245\u001b[0m \u001b[43m \u001b[49m\u001b[43mtoken\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtoken\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1246\u001b[0m \u001b[43m \u001b[49m\u001b[38;5;66;43;03m# Additional options\u001b[39;49;00m\n\u001b[0;32m 1247\u001b[0m \u001b[43m \u001b[49m\u001b[43mlocal_files_only\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mlocal_files_only\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1248\u001b[0m \u001b[43m \u001b[49m\u001b[43mforce_download\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mforce_download\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1249\u001b[0m \u001b[43m \u001b[49m\u001b[43m)\u001b[49m\n",
|
43 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\huggingface_hub\\file_download.py:1295\u001b[0m, in \u001b[0;36m_hf_hub_download_to_cache_dir\u001b[1;34m(cache_dir, repo_id, filename, repo_type, revision, endpoint, etag_timeout, headers, proxies, token, local_files_only, force_download)\u001b[0m\n\u001b[0;32m 1293\u001b[0m \u001b[38;5;66;03m# Try to get metadata (etag, commit_hash, url, size) from the server.\u001b[39;00m\n\u001b[0;32m 1294\u001b[0m \u001b[38;5;66;03m# If we can't, a HEAD request error is returned.\u001b[39;00m\n\u001b[1;32m-> 1295\u001b[0m (url_to_download, etag, commit_hash, expected_size, head_call_error) \u001b[38;5;241m=\u001b[39m \u001b[43m_get_metadata_or_catch_error\u001b[49m\u001b[43m(\u001b[49m\n\u001b[0;32m 1296\u001b[0m \u001b[43m \u001b[49m\u001b[43mrepo_id\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrepo_id\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1297\u001b[0m \u001b[43m \u001b[49m\u001b[43mfilename\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mfilename\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1298\u001b[0m \u001b[43m \u001b[49m\u001b[43mrepo_type\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrepo_type\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1299\u001b[0m \u001b[43m \u001b[49m\u001b[43mrevision\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrevision\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1300\u001b[0m \u001b[43m \u001b[49m\u001b[43mendpoint\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mendpoint\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1301\u001b[0m \u001b[43m \u001b[49m\u001b[43mproxies\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mproxies\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1302\u001b[0m \u001b[43m \u001b[49m\u001b[43metag_timeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43metag_timeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1303\u001b[0m \u001b[43m \u001b[49m\u001b[43mheaders\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1304\u001b[0m \u001b[43m \u001b[49m\u001b[43mtoken\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtoken\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1305\u001b[0m \u001b[43m \u001b[49m\u001b[43mlocal_files_only\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mlocal_files_only\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1306\u001b[0m \u001b[43m \u001b[49m\u001b[43mstorage_folder\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mstorage_folder\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1307\u001b[0m \u001b[43m \u001b[49m\u001b[43mrelative_filename\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrelative_filename\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1308\u001b[0m \u001b[43m\u001b[49m\u001b[43m)\u001b[49m\n\u001b[0;32m 1310\u001b[0m \u001b[38;5;66;03m# etag can be None for several reasons:\u001b[39;00m\n\u001b[0;32m 1311\u001b[0m \u001b[38;5;66;03m# 1. we passed local_files_only.\u001b[39;00m\n\u001b[0;32m 1312\u001b[0m \u001b[38;5;66;03m# 2. we don't have a connection\u001b[39;00m\n\u001b[1;32m (...)\u001b[0m\n\u001b[0;32m 1318\u001b[0m \u001b[38;5;66;03m# If the specified revision is a commit hash, look inside \"snapshots\".\u001b[39;00m\n\u001b[0;32m 1319\u001b[0m \u001b[38;5;66;03m# If the specified revision is a branch or tag, look inside \"refs\".\u001b[39;00m\n",
|
44 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\huggingface_hub\\file_download.py:1746\u001b[0m, in \u001b[0;36m_get_metadata_or_catch_error\u001b[1;34m(repo_id, filename, repo_type, revision, endpoint, proxies, etag_timeout, headers, token, local_files_only, relative_filename, storage_folder)\u001b[0m\n\u001b[0;32m 1745\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[1;32m-> 1746\u001b[0m metadata \u001b[38;5;241m=\u001b[39m \u001b[43mget_hf_file_metadata\u001b[49m\u001b[43m(\u001b[49m\n\u001b[0;32m 1747\u001b[0m \u001b[43m \u001b[49m\u001b[43murl\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mproxies\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mproxies\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mtimeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43metag_timeout\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mheaders\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mtoken\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtoken\u001b[49m\n\u001b[0;32m 1748\u001b[0m \u001b[43m \u001b[49m\u001b[43m)\u001b[49m\n\u001b[0;32m 1749\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m EntryNotFoundError \u001b[38;5;28;01mas\u001b[39;00m http_error:\n",
|
45 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\huggingface_hub\\utils\\_validators.py:114\u001b[0m, in \u001b[0;36mvalidate_hf_hub_args.<locals>._inner_fn\u001b[1;34m(*args, **kwargs)\u001b[0m\n\u001b[0;32m 112\u001b[0m kwargs \u001b[38;5;241m=\u001b[39m smoothly_deprecate_use_auth_token(fn_name\u001b[38;5;241m=\u001b[39mfn\u001b[38;5;241m.\u001b[39m\u001b[38;5;18m__name__\u001b[39m, has_token\u001b[38;5;241m=\u001b[39mhas_token, kwargs\u001b[38;5;241m=\u001b[39mkwargs)\n\u001b[1;32m--> 114\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m fn(\u001b[38;5;241m*\u001b[39margs, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs)\n",
|
46 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\huggingface_hub\\file_download.py:1666\u001b[0m, in \u001b[0;36mget_hf_file_metadata\u001b[1;34m(url, token, proxies, timeout, library_name, library_version, user_agent, headers)\u001b[0m\n\u001b[0;32m 1665\u001b[0m \u001b[38;5;66;03m# Retrieve metadata\u001b[39;00m\n\u001b[1;32m-> 1666\u001b[0m r \u001b[38;5;241m=\u001b[39m \u001b[43m_request_wrapper\u001b[49m\u001b[43m(\u001b[49m\n\u001b[0;32m 1667\u001b[0m \u001b[43m \u001b[49m\u001b[43mmethod\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mHEAD\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1668\u001b[0m \u001b[43m \u001b[49m\u001b[43murl\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1669\u001b[0m \u001b[43m \u001b[49m\u001b[43mheaders\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1670\u001b[0m \u001b[43m \u001b[49m\u001b[43mallow_redirects\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mFalse\u001b[39;49;00m\u001b[43m,\u001b[49m\n\u001b[0;32m 1671\u001b[0m \u001b[43m \u001b[49m\u001b[43mfollow_relative_redirects\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mTrue\u001b[39;49;00m\u001b[43m,\u001b[49m\n\u001b[0;32m 1672\u001b[0m \u001b[43m \u001b[49m\u001b[43mproxies\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mproxies\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1673\u001b[0m \u001b[43m \u001b[49m\u001b[43mtimeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtimeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 1674\u001b[0m \u001b[43m\u001b[49m\u001b[43m)\u001b[49m\n\u001b[0;32m 1675\u001b[0m hf_raise_for_status(r)\n",
|
47 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\huggingface_hub\\file_download.py:364\u001b[0m, in \u001b[0;36m_request_wrapper\u001b[1;34m(method, url, follow_relative_redirects, **params)\u001b[0m\n\u001b[0;32m 363\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m follow_relative_redirects:\n\u001b[1;32m--> 364\u001b[0m response \u001b[38;5;241m=\u001b[39m _request_wrapper(\n\u001b[0;32m 365\u001b[0m method\u001b[38;5;241m=\u001b[39mmethod,\n\u001b[0;32m 366\u001b[0m url\u001b[38;5;241m=\u001b[39murl,\n\u001b[0;32m 367\u001b[0m follow_relative_redirects\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mFalse\u001b[39;00m,\n\u001b[0;32m 368\u001b[0m \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mparams,\n\u001b[0;32m 369\u001b[0m )\n\u001b[0;32m 371\u001b[0m \u001b[38;5;66;03m# If redirection, we redirect only relative paths.\u001b[39;00m\n\u001b[0;32m 372\u001b[0m \u001b[38;5;66;03m# This is useful in case of a renamed repository.\u001b[39;00m\n",
|
48 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\huggingface_hub\\file_download.py:388\u001b[0m, in \u001b[0;36m_request_wrapper\u001b[1;34m(method, url, follow_relative_redirects, **params)\u001b[0m\n\u001b[0;32m 387\u001b[0m response \u001b[38;5;241m=\u001b[39m get_session()\u001b[38;5;241m.\u001b[39mrequest(method\u001b[38;5;241m=\u001b[39mmethod, url\u001b[38;5;241m=\u001b[39murl, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mparams)\n\u001b[1;32m--> 388\u001b[0m \u001b[43mhf_raise_for_status\u001b[49m\u001b[43m(\u001b[49m\u001b[43mresponse\u001b[49m\u001b[43m)\u001b[49m\n\u001b[0;32m 389\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m response\n",
|
49 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\huggingface_hub\\utils\\_http.py:417\u001b[0m, in \u001b[0;36mhf_raise_for_status\u001b[1;34m(response, endpoint_name)\u001b[0m\n\u001b[0;32m 416\u001b[0m message \u001b[38;5;241m=\u001b[39m \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;132;01m{\u001b[39;00mresponse\u001b[38;5;241m.\u001b[39mstatus_code\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m Client Error.\u001b[39m\u001b[38;5;124m\"\u001b[39m \u001b[38;5;241m+\u001b[39m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;124m\"\u001b[39m \u001b[38;5;241m+\u001b[39m \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mEntry Not Found for url: \u001b[39m\u001b[38;5;132;01m{\u001b[39;00mresponse\u001b[38;5;241m.\u001b[39murl\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m.\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[1;32m--> 417\u001b[0m \u001b[38;5;28;01mraise\u001b[39;00m _format(EntryNotFoundError, message, response) \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01me\u001b[39;00m\n\u001b[0;32m 419\u001b[0m \u001b[38;5;28;01melif\u001b[39;00m error_code \u001b[38;5;241m==\u001b[39m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mGatedRepo\u001b[39m\u001b[38;5;124m\"\u001b[39m:\n",
|
50 |
+
"\u001b[1;31mEntryNotFoundError\u001b[0m: 404 Client Error. (Request ID: Root=1-679b5d06-582a8eae4ac3a65307600329;86a3a5c8-5d12-4e82-87db-9c59aee9335c)\n\nEntry Not Found for url: https://huggingface.co/AniMAntZeZo/RuBert-tiny2-EmotionsDetected/resolve/main/config.json.",
|
51 |
+
"\nThe above exception was the direct cause of the following exception:\n",
|
52 |
+
"\u001b[1;31mOSError\u001b[0m Traceback (most recent call last)",
|
53 |
+
"Cell \u001b[1;32mIn[2], line 1\u001b[0m\n\u001b[1;32m----> 1\u001b[0m tokenizer \u001b[38;5;241m=\u001b[39m \u001b[43mAutoTokenizer\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mfrom_pretrained\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;124;43m'\u001b[39;49m\u001b[38;5;124;43mAniMAntZeZo/RuBert-tiny2-EmotionsDetected\u001b[39;49m\u001b[38;5;124;43m'\u001b[39;49m\u001b[43m)\u001b[49m\n\u001b[0;32m 2\u001b[0m model \u001b[38;5;241m=\u001b[39m AutoModelForSequenceClassification\u001b[38;5;241m.\u001b[39mfrom_pretrained(\u001b[38;5;124m'\u001b[39m\u001b[38;5;124mAniMAntZeZo/RuBert-tiny2-EmotionsDetected\u001b[39m\u001b[38;5;124m'\u001b[39m)\n\u001b[0;32m 3\u001b[0m model\u001b[38;5;241m.\u001b[39mto(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mcuda\u001b[39m\u001b[38;5;124m\"\u001b[39m \u001b[38;5;28;01mif\u001b[39;00m torch\u001b[38;5;241m.\u001b[39mcuda\u001b[38;5;241m.\u001b[39mis_available() \u001b[38;5;28;01melse\u001b[39;00m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mcpu\u001b[39m\u001b[38;5;124m\"\u001b[39m, \n\u001b[0;32m 4\u001b[0m )\n",
|
54 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\transformers\\models\\auto\\tokenization_auto.py:773\u001b[0m, in \u001b[0;36mAutoTokenizer.from_pretrained\u001b[1;34m(cls, pretrained_model_name_or_path, *inputs, **kwargs)\u001b[0m\n\u001b[0;32m 771\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m config_tokenizer_class \u001b[38;5;129;01mis\u001b[39;00m \u001b[38;5;28;01mNone\u001b[39;00m:\n\u001b[0;32m 772\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;129;01mnot\u001b[39;00m \u001b[38;5;28misinstance\u001b[39m(config, PretrainedConfig):\n\u001b[1;32m--> 773\u001b[0m config \u001b[38;5;241m=\u001b[39m AutoConfig\u001b[38;5;241m.\u001b[39mfrom_pretrained(\n\u001b[0;32m 774\u001b[0m pretrained_model_name_or_path, trust_remote_code\u001b[38;5;241m=\u001b[39mtrust_remote_code, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs\n\u001b[0;32m 775\u001b[0m )\n\u001b[0;32m 776\u001b[0m config_tokenizer_class \u001b[38;5;241m=\u001b[39m config\u001b[38;5;241m.\u001b[39mtokenizer_class\n\u001b[0;32m 777\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28mhasattr\u001b[39m(config, \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mauto_map\u001b[39m\u001b[38;5;124m\"\u001b[39m) \u001b[38;5;129;01mand\u001b[39;00m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mAutoTokenizer\u001b[39m\u001b[38;5;124m\"\u001b[39m \u001b[38;5;129;01min\u001b[39;00m config\u001b[38;5;241m.\u001b[39mauto_map:\n",
|
55 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\transformers\\models\\auto\\configuration_auto.py:1100\u001b[0m, in \u001b[0;36mAutoConfig.from_pretrained\u001b[1;34m(cls, pretrained_model_name_or_path, **kwargs)\u001b[0m\n\u001b[0;32m 1097\u001b[0m trust_remote_code \u001b[38;5;241m=\u001b[39m kwargs\u001b[38;5;241m.\u001b[39mpop(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mtrust_remote_code\u001b[39m\u001b[38;5;124m\"\u001b[39m, \u001b[38;5;28;01mNone\u001b[39;00m)\n\u001b[0;32m 1098\u001b[0m code_revision \u001b[38;5;241m=\u001b[39m kwargs\u001b[38;5;241m.\u001b[39mpop(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mcode_revision\u001b[39m\u001b[38;5;124m\"\u001b[39m, \u001b[38;5;28;01mNone\u001b[39;00m)\n\u001b[1;32m-> 1100\u001b[0m config_dict, unused_kwargs \u001b[38;5;241m=\u001b[39m PretrainedConfig\u001b[38;5;241m.\u001b[39mget_config_dict(pretrained_model_name_or_path, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs)\n\u001b[0;32m 1101\u001b[0m has_remote_code \u001b[38;5;241m=\u001b[39m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mauto_map\u001b[39m\u001b[38;5;124m\"\u001b[39m \u001b[38;5;129;01min\u001b[39;00m config_dict \u001b[38;5;129;01mand\u001b[39;00m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mAutoConfig\u001b[39m\u001b[38;5;124m\"\u001b[39m \u001b[38;5;129;01min\u001b[39;00m config_dict[\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mauto_map\u001b[39m\u001b[38;5;124m\"\u001b[39m]\n\u001b[0;32m 1102\u001b[0m has_local_code \u001b[38;5;241m=\u001b[39m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mmodel_type\u001b[39m\u001b[38;5;124m\"\u001b[39m \u001b[38;5;129;01min\u001b[39;00m config_dict \u001b[38;5;129;01mand\u001b[39;00m config_dict[\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mmodel_type\u001b[39m\u001b[38;5;124m\"\u001b[39m] \u001b[38;5;129;01min\u001b[39;00m CONFIG_MAPPING\n",
|
56 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\transformers\\configuration_utils.py:634\u001b[0m, in \u001b[0;36mPretrainedConfig.get_config_dict\u001b[1;34m(cls, pretrained_model_name_or_path, **kwargs)\u001b[0m\n\u001b[0;32m 632\u001b[0m original_kwargs \u001b[38;5;241m=\u001b[39m copy\u001b[38;5;241m.\u001b[39mdeepcopy(kwargs)\n\u001b[0;32m 633\u001b[0m \u001b[38;5;66;03m# Get config dict associated with the base config file\u001b[39;00m\n\u001b[1;32m--> 634\u001b[0m config_dict, kwargs \u001b[38;5;241m=\u001b[39m \u001b[38;5;28mcls\u001b[39m\u001b[38;5;241m.\u001b[39m_get_config_dict(pretrained_model_name_or_path, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs)\n\u001b[0;32m 635\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124m_commit_hash\u001b[39m\u001b[38;5;124m\"\u001b[39m \u001b[38;5;129;01min\u001b[39;00m config_dict:\n\u001b[0;32m 636\u001b[0m original_kwargs[\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124m_commit_hash\u001b[39m\u001b[38;5;124m\"\u001b[39m] \u001b[38;5;241m=\u001b[39m config_dict[\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124m_commit_hash\u001b[39m\u001b[38;5;124m\"\u001b[39m]\n",
|
57 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\transformers\\configuration_utils.py:689\u001b[0m, in \u001b[0;36mPretrainedConfig._get_config_dict\u001b[1;34m(cls, pretrained_model_name_or_path, **kwargs)\u001b[0m\n\u001b[0;32m 685\u001b[0m configuration_file \u001b[38;5;241m=\u001b[39m kwargs\u001b[38;5;241m.\u001b[39mpop(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124m_configuration_file\u001b[39m\u001b[38;5;124m\"\u001b[39m, CONFIG_NAME)\n\u001b[0;32m 687\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m 688\u001b[0m \u001b[38;5;66;03m# Load from local folder or from cache or download from model Hub and cache\u001b[39;00m\n\u001b[1;32m--> 689\u001b[0m resolved_config_file \u001b[38;5;241m=\u001b[39m \u001b[43mcached_file\u001b[49m\u001b[43m(\u001b[49m\n\u001b[0;32m 690\u001b[0m \u001b[43m \u001b[49m\u001b[43mpretrained_model_name_or_path\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 691\u001b[0m \u001b[43m \u001b[49m\u001b[43mconfiguration_file\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 692\u001b[0m \u001b[43m \u001b[49m\u001b[43mcache_dir\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mcache_dir\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 693\u001b[0m \u001b[43m \u001b[49m\u001b[43mforce_download\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mforce_download\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 694\u001b[0m \u001b[43m \u001b[49m\u001b[43mproxies\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mproxies\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 695\u001b[0m \u001b[43m \u001b[49m\u001b[43mresume_download\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mresume_download\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 696\u001b[0m \u001b[43m \u001b[49m\u001b[43mlocal_files_only\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mlocal_files_only\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 697\u001b[0m \u001b[43m \u001b[49m\u001b[43mtoken\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtoken\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 698\u001b[0m \u001b[43m \u001b[49m\u001b[43muser_agent\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43muser_agent\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 699\u001b[0m \u001b[43m \u001b[49m\u001b[43mrevision\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrevision\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 700\u001b[0m \u001b[43m \u001b[49m\u001b[43msubfolder\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43msubfolder\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 701\u001b[0m \u001b[43m \u001b[49m\u001b[43m_commit_hash\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mcommit_hash\u001b[49m\u001b[43m,\u001b[49m\n\u001b[0;32m 702\u001b[0m \u001b[43m \u001b[49m\u001b[43m)\u001b[49m\n\u001b[0;32m 703\u001b[0m commit_hash \u001b[38;5;241m=\u001b[39m extract_commit_hash(resolved_config_file, commit_hash)\n\u001b[0;32m 704\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m \u001b[38;5;167;01mEnvironmentError\u001b[39;00m:\n\u001b[0;32m 705\u001b[0m \u001b[38;5;66;03m# Raise any environment error raise by `cached_file`. It will have a helpful error message adapted to\u001b[39;00m\n\u001b[0;32m 706\u001b[0m \u001b[38;5;66;03m# the original exception.\u001b[39;00m\n",
|
58 |
+
"File \u001b[1;32mc:\\Users\\Admin\\AppData\\Local\\Programs\\Python\\Python310\\lib\\site-packages\\transformers\\utils\\hub.py:436\u001b[0m, in \u001b[0;36mcached_file\u001b[1;34m(path_or_repo_id, filename, cache_dir, force_download, resume_download, proxies, token, revision, local_files_only, subfolder, repo_type, user_agent, _raise_exceptions_for_missing_entries, _raise_exceptions_for_connection_errors, _commit_hash, **deprecated_kwargs)\u001b[0m\n\u001b[0;32m 434\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m revision \u001b[38;5;129;01mis\u001b[39;00m \u001b[38;5;28;01mNone\u001b[39;00m:\n\u001b[0;32m 435\u001b[0m revision \u001b[38;5;241m=\u001b[39m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mmain\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[1;32m--> 436\u001b[0m \u001b[38;5;28;01mraise\u001b[39;00m \u001b[38;5;167;01mEnvironmentError\u001b[39;00m(\n\u001b[0;32m 437\u001b[0m \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;132;01m{\u001b[39;00mpath_or_repo_id\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m does not appear to have a file named \u001b[39m\u001b[38;5;132;01m{\u001b[39;00mfull_filename\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m. Checkout \u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[0;32m 438\u001b[0m \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124m'\u001b[39m\u001b[38;5;124mhttps://huggingface.co/\u001b[39m\u001b[38;5;132;01m{\u001b[39;00mpath_or_repo_id\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m/\u001b[39m\u001b[38;5;132;01m{\u001b[39;00mrevision\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m'\u001b[39m\u001b[38;5;124m for available files.\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[0;32m 439\u001b[0m ) \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01me\u001b[39;00m\n\u001b[0;32m 440\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m HTTPError \u001b[38;5;28;01mas\u001b[39;00m err:\n\u001b[0;32m 441\u001b[0m \u001b[38;5;66;03m# First we try to see if we have a cached version (not up to date):\u001b[39;00m\n\u001b[0;32m 442\u001b[0m resolved_file \u001b[38;5;241m=\u001b[39m try_to_load_from_cache(path_or_repo_id, full_filename, cache_dir\u001b[38;5;241m=\u001b[39mcache_dir, revision\u001b[38;5;241m=\u001b[39mrevision)\n",
|
59 |
+
"\u001b[1;31mOSError\u001b[0m: AniMAntZeZo/RuBert-tiny2-EmotionsDetected does not appear to have a file named config.json. Checkout 'https://huggingface.co/AniMAntZeZo/RuBert-tiny2-EmotionsDetected/main' for available files."
|
60 |
+
]
|
61 |
+
}
|
62 |
+
],
|
63 |
+
"source": [
|
64 |
+
"tokenizer = AutoTokenizer.from_pretrained('AniMAntZeZo/RuBert-tiny2-EmotionsDetected')\n",
|
65 |
+
"model = AutoModelForSequenceClassification.from_pretrained('AniMAntZeZo/RuBert-tiny2-EmotionsDetected')\n",
|
66 |
+
"model.to(\"cuda\" if torch.cuda.is_available() else \"cpu\", \n",
|
67 |
+
")"
|
68 |
+
]
|
69 |
+
},
|
70 |
+
{
|
71 |
+
"cell_type": "code",
|
72 |
+
"execution_count": null,
|
73 |
+
"metadata": {},
|
74 |
+
"outputs": [],
|
75 |
+
"source": [
|
76 |
+
"emotion_columns = [\n",
|
77 |
+
" \"admiration\", \"amusement\", \"anger\", \"annoyance\", \"approval\", \"caring\", \"confusion\", \"curiosity\", \"desire\",\n",
|
78 |
+
" \"disappointment\", \"disapproval\", \"disgust\", \"embarrassment\", \"excitement\", \"fear\", \"gratitude\", \"grief\", \"joy\",\n",
|
79 |
+
" \"love\", \"nervousness\", \"optimism\", \"pride\", \"realization\", \"relief\", \"remorse\", \"sadness\", \"surprise\", \"neutral\"\n",
|
80 |
+
"]"
|
81 |
+
]
|
82 |
+
},
|
83 |
+
{
|
84 |
+
"cell_type": "code",
|
85 |
+
"execution_count": null,
|
86 |
+
"metadata": {},
|
87 |
+
"outputs": [],
|
88 |
+
"source": [
|
89 |
+
"def predict_emotions(\n",
|
90 |
+
" text, \n",
|
91 |
+
" model, \n",
|
92 |
+
" tokenizer, \n",
|
93 |
+
" emotion_columns, \n",
|
94 |
+
" device=\"cuda\" if torch.cuda.is_available() else \"cpu\", \n",
|
95 |
+
" threshold=0.1\n",
|
96 |
+
"):\n",
|
97 |
+
"\n",
|
98 |
+
" emotion_translations = {\n",
|
99 |
+
" \"admiration\": \"восхищение\",\n",
|
100 |
+
" \"amusement\": \"веселье\",\n",
|
101 |
+
" \"anger\": \"злость\",\n",
|
102 |
+
" \"annoyance\": \"раздражение\",\n",
|
103 |
+
" \"approval\": \"одобрение\",\n",
|
104 |
+
" \"caring\": \"забота\",\n",
|
105 |
+
" \"confusion\": \"непонимание\",\n",
|
106 |
+
" \"curiosity\": \"любопытство\",\n",
|
107 |
+
" \"desire\": \"желание\",\n",
|
108 |
+
" \"disappointment\": \"разочарование\",\n",
|
109 |
+
" \"disapproval\": \"неодобрение\",\n",
|
110 |
+
" \"disgust\": \"отвращение\",\n",
|
111 |
+
" \"embarrassment\": \"смущение\",\n",
|
112 |
+
" \"excitement\": \"возбуждение\",\n",
|
113 |
+
" \"fear\": \"страх\",\n",
|
114 |
+
" \"gratitude\": \"признательность\",\n",
|
115 |
+
" \"grief\": \"горе\",\n",
|
116 |
+
" \"joy\": \"радость\",\n",
|
117 |
+
" \"love\": \"любовь\",\n",
|
118 |
+
" \"nervousness\": \"нервозность\",\n",
|
119 |
+
" \"optimism\": \"оптимизм\",\n",
|
120 |
+
" \"pride\": \"гордость\",\n",
|
121 |
+
" \"realization\": \"осознание\",\n",
|
122 |
+
" \"relief\": \"облегчение\",\n",
|
123 |
+
" \"remorse\": \"раскаяние\",\n",
|
124 |
+
" \"sadness\": \"грусть\",\n",
|
125 |
+
" \"surprise\": \"удивление\",\n",
|
126 |
+
" \"neutral\": \"нейтральность\",\n",
|
127 |
+
" }\n",
|
128 |
+
"\n",
|
129 |
+
" model.to(device)\n",
|
130 |
+
" model.eval()\n",
|
131 |
+
" inputs = tokenizer(text, return_tensors=\"pt\", padding=\"max_length\", truncation=True, max_length=128).to(device)\n",
|
132 |
+
" with torch.no_grad():\n",
|
133 |
+
" logits = model(**inputs).logits\n",
|
134 |
+
" probabilities = torch.sigmoid(logits).squeeze().cpu().numpy()\n",
|
135 |
+
"\n",
|
136 |
+
" predictions = {\n",
|
137 |
+
" f\"{emotion} ({emotion_translations[emotion]})\": prob\n",
|
138 |
+
" for emotion, prob in zip(emotion_columns, probabilities) if prob > threshold\n",
|
139 |
+
" }\n",
|
140 |
+
"\n",
|
141 |
+
" sorted_predictions = dict(sorted(predictions.items(), key=lambda item: item[1], reverse=True))\n",
|
142 |
+
" \n",
|
143 |
+
" return sorted_predictions\n"
|
144 |
+
]
|
145 |
+
},
|
146 |
+
{
|
147 |
+
"cell_type": "code",
|
148 |
+
"execution_count": null,
|
149 |
+
"metadata": {},
|
150 |
+
"outputs": [],
|
151 |
+
"source": [
|
152 |
+
"example_text = \"Как же я рад!\"\n",
|
153 |
+
"predictions = predict_emotions(example_text, model, tokenizer, emotion_columns)\n",
|
154 |
+
"print(\"Emotions:\", predictions)"
|
155 |
+
]
|
156 |
+
}
|
157 |
+
],
|
158 |
+
"metadata": {
|
159 |
+
"kernelspec": {
|
160 |
+
"display_name": "Python 3",
|
161 |
+
"language": "python",
|
162 |
+
"name": "python3"
|
163 |
+
},
|
164 |
+
"language_info": {
|
165 |
+
"codemirror_mode": {
|
166 |
+
"name": "ipython",
|
167 |
+
"version": 3
|
168 |
+
},
|
169 |
+
"file_extension": ".py",
|
170 |
+
"mimetype": "text/x-python",
|
171 |
+
"name": "python",
|
172 |
+
"nbconvert_exporter": "python",
|
173 |
+
"pygments_lexer": "ipython3",
|
174 |
+
"version": "3.10.0rc2"
|
175 |
+
}
|
176 |
+
},
|
177 |
+
"nbformat": 4,
|
178 |
+
"nbformat_minor": 2
|
179 |
+
}
|
tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
tokenizer_config.json
ADDED
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"added_tokens_decoder": {
|
3 |
+
"0": {
|
4 |
+
"content": "[PAD]",
|
5 |
+
"lstrip": false,
|
6 |
+
"normalized": false,
|
7 |
+
"rstrip": false,
|
8 |
+
"single_word": false,
|
9 |
+
"special": true
|
10 |
+
},
|
11 |
+
"1": {
|
12 |
+
"content": "[UNK]",
|
13 |
+
"lstrip": false,
|
14 |
+
"normalized": false,
|
15 |
+
"rstrip": false,
|
16 |
+
"single_word": false,
|
17 |
+
"special": true
|
18 |
+
},
|
19 |
+
"2": {
|
20 |
+
"content": "[CLS]",
|
21 |
+
"lstrip": false,
|
22 |
+
"normalized": false,
|
23 |
+
"rstrip": false,
|
24 |
+
"single_word": false,
|
25 |
+
"special": true
|
26 |
+
},
|
27 |
+
"3": {
|
28 |
+
"content": "[SEP]",
|
29 |
+
"lstrip": false,
|
30 |
+
"normalized": false,
|
31 |
+
"rstrip": false,
|
32 |
+
"single_word": false,
|
33 |
+
"special": true
|
34 |
+
},
|
35 |
+
"4": {
|
36 |
+
"content": "[MASK]",
|
37 |
+
"lstrip": false,
|
38 |
+
"normalized": false,
|
39 |
+
"rstrip": false,
|
40 |
+
"single_word": false,
|
41 |
+
"special": true
|
42 |
+
}
|
43 |
+
},
|
44 |
+
"clean_up_tokenization_spaces": false,
|
45 |
+
"cls_token": "[CLS]",
|
46 |
+
"do_basic_tokenize": true,
|
47 |
+
"do_lower_case": false,
|
48 |
+
"extra_special_tokens": {},
|
49 |
+
"mask_token": "[MASK]",
|
50 |
+
"max_length": 512,
|
51 |
+
"model_max_length": 2048,
|
52 |
+
"never_split": null,
|
53 |
+
"pad_to_multiple_of": null,
|
54 |
+
"pad_token": "[PAD]",
|
55 |
+
"pad_token_type_id": 0,
|
56 |
+
"padding_side": "right",
|
57 |
+
"sep_token": "[SEP]",
|
58 |
+
"stride": 0,
|
59 |
+
"strip_accents": null,
|
60 |
+
"tokenize_chinese_chars": true,
|
61 |
+
"tokenizer_class": "BertTokenizer",
|
62 |
+
"truncation_side": "right",
|
63 |
+
"truncation_strategy": "longest_first",
|
64 |
+
"unk_token": "[UNK]"
|
65 |
+
}
|
vocab.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|