AniMAntZeZo commited on
Commit
c7c2507
·
verified ·
1 Parent(s): c9276bf

Upload 20 files

Browse files
.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