AndreMitri commited on
Commit
c1f801a
β€’
1 Parent(s): 1ba6bc3

Notebooks Explicativos, imagens e dados

Browse files

Adiciona
- Notebooks Explicativos
- Imagens
- Dados

.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip 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
 
 
33
  *.zip 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
+ data/imdb_reviews.csv filter=lfs diff=lfs merge=lfs -text
data/imdb_reviews.csv ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f1314f123ac922d7d0f2bd5bd17f1734e167d90b2256c34963228bc63f6a4cb
3
+ size 66262310
imagens/BERT_TDIDF.png ADDED
imagens/Simbolico_WordCloud_Wordnet.png ADDED
notebooks_explicativos/Estatistico.ipynb ADDED
The diff for this file is too large to render. See raw diff
 
notebooks_explicativos/Neural_Bert.ipynb ADDED
@@ -0,0 +1,1291 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cells": [
3
+ {
4
+ "cell_type": "markdown",
5
+ "metadata": {},
6
+ "source": [
7
+ "# SCC0633/SCC5908 - Processamento de Linguagem Natural\n",
8
+ "> **Docente:** Thiago Alexandre Salgueiro Pardo \\\\\n",
9
+ "> **EstagiΓ‘rio PAE:** Germano Antonio Zani Jorge\n",
10
+ "\n",
11
+ "\n",
12
+ "# Integrantes do Grupo: GPTrouxas\n",
13
+ "> AndrΓ© Guarnier De Mitri - 11395579 \\\\\n",
14
+ "> Daniel Carvalho - 10685702 \\\\\n",
15
+ "> Fernando - 11795342 \\\\\n",
16
+ "> Lucas Henrique Sant'Anna - 10748521 \\\\\n",
17
+ "> Magaly L Fujimoto - 4890582 \\\\\n"
18
+ ]
19
+ },
20
+ {
21
+ "cell_type": "markdown",
22
+ "metadata": {},
23
+ "source": [
24
+ "# Abordagem Neural usando BERT\n",
25
+ "![alt text](../imagens/BERT_TDIDF.png)"
26
+ ]
27
+ },
28
+ {
29
+ "cell_type": "markdown",
30
+ "metadata": {},
31
+ "source": [
32
+ "###"
33
+ ]
34
+ },
35
+ {
36
+ "cell_type": "markdown",
37
+ "metadata": {
38
+ "id": "6yecpJR0feeQ"
39
+ },
40
+ "source": [
41
+ "## Importando bibliotecas"
42
+ ]
43
+ },
44
+ {
45
+ "cell_type": "code",
46
+ "execution_count": 1,
47
+ "metadata": {
48
+ "id": "FAIvyZwodEtm"
49
+ },
50
+ "outputs": [],
51
+ "source": [
52
+ "import torch\n",
53
+ "import numpy as np\n",
54
+ "import matplotlib.pyplot as plt\n",
55
+ "import math\n",
56
+ "from tqdm.notebook import tqdm\n",
57
+ "import pandas as pd"
58
+ ]
59
+ },
60
+ {
61
+ "cell_type": "code",
62
+ "execution_count": 3,
63
+ "metadata": {},
64
+ "outputs": [],
65
+ "source": [
66
+ "#!pip install transformers seaborn nltk"
67
+ ]
68
+ },
69
+ {
70
+ "cell_type": "markdown",
71
+ "metadata": {},
72
+ "source": [
73
+ "## Carregando dados"
74
+ ]
75
+ },
76
+ {
77
+ "cell_type": "code",
78
+ "execution_count": 3,
79
+ "metadata": {
80
+ "colab": {
81
+ "base_uri": "https://localhost:8080/",
82
+ "height": 206
83
+ },
84
+ "id": "LYgXl3RIfgfo",
85
+ "outputId": "eb496faf-7826-44f7-fa88-3b21fb6e7cbf"
86
+ },
87
+ "outputs": [
88
+ {
89
+ "data": {
90
+ "text/html": [
91
+ "<div>\n",
92
+ "<style scoped>\n",
93
+ " .dataframe tbody tr th:only-of-type {\n",
94
+ " vertical-align: middle;\n",
95
+ " }\n",
96
+ "\n",
97
+ " .dataframe tbody tr th {\n",
98
+ " vertical-align: top;\n",
99
+ " }\n",
100
+ "\n",
101
+ " .dataframe thead th {\n",
102
+ " text-align: right;\n",
103
+ " }\n",
104
+ "</style>\n",
105
+ "<table border=\"1\" class=\"dataframe\">\n",
106
+ " <thead>\n",
107
+ " <tr style=\"text-align: right;\">\n",
108
+ " <th></th>\n",
109
+ " <th>review</th>\n",
110
+ " <th>sentiment</th>\n",
111
+ " </tr>\n",
112
+ " </thead>\n",
113
+ " <tbody>\n",
114
+ " <tr>\n",
115
+ " <th>0</th>\n",
116
+ " <td>One of the other reviewers has mentioned that ...</td>\n",
117
+ " <td>positive</td>\n",
118
+ " </tr>\n",
119
+ " <tr>\n",
120
+ " <th>1</th>\n",
121
+ " <td>A wonderful little production. &lt;br /&gt;&lt;br /&gt;The...</td>\n",
122
+ " <td>positive</td>\n",
123
+ " </tr>\n",
124
+ " <tr>\n",
125
+ " <th>2</th>\n",
126
+ " <td>I thought this was a wonderful way to spend ti...</td>\n",
127
+ " <td>positive</td>\n",
128
+ " </tr>\n",
129
+ " <tr>\n",
130
+ " <th>3</th>\n",
131
+ " <td>Basically there's a family where a little boy ...</td>\n",
132
+ " <td>negative</td>\n",
133
+ " </tr>\n",
134
+ " <tr>\n",
135
+ " <th>4</th>\n",
136
+ " <td>Petter Mattei's \"Love in the Time of Money\" is...</td>\n",
137
+ " <td>positive</td>\n",
138
+ " </tr>\n",
139
+ " </tbody>\n",
140
+ "</table>\n",
141
+ "</div>"
142
+ ],
143
+ "text/plain": [
144
+ " review sentiment\n",
145
+ "0 One of the other reviewers has mentioned that ... positive\n",
146
+ "1 A wonderful little production. <br /><br />The... positive\n",
147
+ "2 I thought this was a wonderful way to spend ti... positive\n",
148
+ "3 Basically there's a family where a little boy ... negative\n",
149
+ "4 Petter Mattei's \"Love in the Time of Money\" is... positive"
150
+ ]
151
+ },
152
+ "execution_count": 3,
153
+ "metadata": {},
154
+ "output_type": "execute_result"
155
+ }
156
+ ],
157
+ "source": [
158
+ "df_reviews = pd.read_csv('imdb_reviews.csv')\n",
159
+ "df_reviews.head()"
160
+ ]
161
+ },
162
+ {
163
+ "cell_type": "markdown",
164
+ "metadata": {},
165
+ "source": [
166
+ "## Mapeando as classes\n",
167
+ "- Sentimento positivo recebe label 1\n",
168
+ "- Sentimento negativo recebe label 0"
169
+ ]
170
+ },
171
+ {
172
+ "cell_type": "code",
173
+ "execution_count": 4,
174
+ "metadata": {
175
+ "colab": {
176
+ "base_uri": "https://localhost:8080/",
177
+ "height": 206
178
+ },
179
+ "id": "D-5n8XzJbWOO",
180
+ "outputId": "cef630cc-b0cc-4598-c53f-d32636bfcd86"
181
+ },
182
+ "outputs": [
183
+ {
184
+ "data": {
185
+ "text/html": [
186
+ "<div>\n",
187
+ "<style scoped>\n",
188
+ " .dataframe tbody tr th:only-of-type {\n",
189
+ " vertical-align: middle;\n",
190
+ " }\n",
191
+ "\n",
192
+ " .dataframe tbody tr th {\n",
193
+ " vertical-align: top;\n",
194
+ " }\n",
195
+ "\n",
196
+ " .dataframe thead th {\n",
197
+ " text-align: right;\n",
198
+ " }\n",
199
+ "</style>\n",
200
+ "<table border=\"1\" class=\"dataframe\">\n",
201
+ " <thead>\n",
202
+ " <tr style=\"text-align: right;\">\n",
203
+ " <th></th>\n",
204
+ " <th>review</th>\n",
205
+ " <th>sentiment</th>\n",
206
+ " </tr>\n",
207
+ " </thead>\n",
208
+ " <tbody>\n",
209
+ " <tr>\n",
210
+ " <th>0</th>\n",
211
+ " <td>One of the other reviewers has mentioned that ...</td>\n",
212
+ " <td>1</td>\n",
213
+ " </tr>\n",
214
+ " <tr>\n",
215
+ " <th>1</th>\n",
216
+ " <td>A wonderful little production. &lt;br /&gt;&lt;br /&gt;The...</td>\n",
217
+ " <td>1</td>\n",
218
+ " </tr>\n",
219
+ " <tr>\n",
220
+ " <th>2</th>\n",
221
+ " <td>I thought this was a wonderful way to spend ti...</td>\n",
222
+ " <td>1</td>\n",
223
+ " </tr>\n",
224
+ " <tr>\n",
225
+ " <th>3</th>\n",
226
+ " <td>Basically there's a family where a little boy ...</td>\n",
227
+ " <td>0</td>\n",
228
+ " </tr>\n",
229
+ " <tr>\n",
230
+ " <th>4</th>\n",
231
+ " <td>Petter Mattei's \"Love in the Time of Money\" is...</td>\n",
232
+ " <td>1</td>\n",
233
+ " </tr>\n",
234
+ " </tbody>\n",
235
+ "</table>\n",
236
+ "</div>"
237
+ ],
238
+ "text/plain": [
239
+ " review sentiment\n",
240
+ "0 One of the other reviewers has mentioned that ... 1\n",
241
+ "1 A wonderful little production. <br /><br />The... 1\n",
242
+ "2 I thought this was a wonderful way to spend ti... 1\n",
243
+ "3 Basically there's a family where a little boy ... 0\n",
244
+ "4 Petter Mattei's \"Love in the Time of Money\" is... 1"
245
+ ]
246
+ },
247
+ "execution_count": 4,
248
+ "metadata": {},
249
+ "output_type": "execute_result"
250
+ }
251
+ ],
252
+ "source": [
253
+ "def map_sentiments(sentiment):\n",
254
+ " if sentiment == 'positive':\n",
255
+ " return 1\n",
256
+ " return 0\n",
257
+ "\n",
258
+ "df_reviews['sentiment'] = df_reviews['sentiment'].apply(map_sentiments)\n",
259
+ "df_reviews.head()"
260
+ ]
261
+ },
262
+ {
263
+ "cell_type": "markdown",
264
+ "metadata": {},
265
+ "source": [
266
+ "# Funçáes para limpeza do texto\n",
267
+ "**lowercase_text(text)** Converte o texto para letras minΓΊsculas para uniformizar o texto.\n",
268
+ "\n",
269
+ "\n",
270
+ "**remove_html(text)** Remove quaisquer tags HTML do texto para limpar dados provenientes de fontes HTML.\n",
271
+ "\n",
272
+ "\n",
273
+ " **remove_url(text)** Remove URLs do texto para eliminar links que podem nΓ£o ser relevantes para a anΓ‘lise de texto.\n",
274
+ "\n",
275
+ "\n",
276
+ "**remove_punctuations(text)** Remove pontuaçáes do texto para simplificar a estrutura do texto, mantendo apenas palavras.\n",
277
+ "\n",
278
+ "**remove_emojis(text)** Remove emojis do texto para evitar caracteres nΓ£o verbais que podem interferir na anΓ‘lise textual.\n",
279
+ "\n",
280
+ "**remove_stop_words(text)** Remove stop words (palavras comuns como \"e\", \"de\", \"o\") que geralmente nΓ£o adicionam valor significativo Γ  anΓ‘lise de texto.\n",
281
+ "\n",
282
+ "**stem_words(text)** Aplica stemming nas palavras do texto, reduzindo-as à sua raiz (por exemplo, \"running\" vira \"run\") para normalizar as variaçáes das palavras.\n",
283
+ "\n",
284
+ "**preprocess_text(text)** Aplica todas as funçáes acima em sequΓͺncia para prΓ©-processar o texto de forma completa, tornando-o mais adequado para anΓ‘lise de texto ou modelagem.\n",
285
+ "\n",
286
+ "\n",
287
+ "\n"
288
+ ]
289
+ },
290
+ {
291
+ "cell_type": "code",
292
+ "execution_count": 5,
293
+ "metadata": {
294
+ "colab": {
295
+ "base_uri": "https://localhost:8080/",
296
+ "height": 241
297
+ },
298
+ "id": "PnFHO62rnWn-",
299
+ "outputId": "17fb6619-fab9-4395-de5d-4c5199e7e45e"
300
+ },
301
+ "outputs": [
302
+ {
303
+ "name": "stderr",
304
+ "output_type": "stream",
305
+ "text": [
306
+ "[nltk_data] Downloading package stopwords to\n",
307
+ "[nltk_data] C:\\Users\\andre\\AppData\\Roaming\\nltk_data...\n",
308
+ "[nltk_data] Package stopwords is already up-to-date!\n"
309
+ ]
310
+ },
311
+ {
312
+ "data": {
313
+ "text/html": [
314
+ "<div>\n",
315
+ "<style scoped>\n",
316
+ " .dataframe tbody tr th:only-of-type {\n",
317
+ " vertical-align: middle;\n",
318
+ " }\n",
319
+ "\n",
320
+ " .dataframe tbody tr th {\n",
321
+ " vertical-align: top;\n",
322
+ " }\n",
323
+ "\n",
324
+ " .dataframe thead th {\n",
325
+ " text-align: right;\n",
326
+ " }\n",
327
+ "</style>\n",
328
+ "<table border=\"1\" class=\"dataframe\">\n",
329
+ " <thead>\n",
330
+ " <tr style=\"text-align: right;\">\n",
331
+ " <th></th>\n",
332
+ " <th>review</th>\n",
333
+ " <th>sentiment</th>\n",
334
+ " </tr>\n",
335
+ " </thead>\n",
336
+ " <tbody>\n",
337
+ " <tr>\n",
338
+ " <th>0</th>\n",
339
+ " <td>one review mention watch 1 oz episod hook righ...</td>\n",
340
+ " <td>1</td>\n",
341
+ " </tr>\n",
342
+ " <tr>\n",
343
+ " <th>1</th>\n",
344
+ " <td>wonder littl product film techniqu unassum old...</td>\n",
345
+ " <td>1</td>\n",
346
+ " </tr>\n",
347
+ " <tr>\n",
348
+ " <th>2</th>\n",
349
+ " <td>thought wonder way spend time hot summer weeke...</td>\n",
350
+ " <td>1</td>\n",
351
+ " </tr>\n",
352
+ " <tr>\n",
353
+ " <th>3</th>\n",
354
+ " <td>basic famili littl boy jake think zombi closet...</td>\n",
355
+ " <td>0</td>\n",
356
+ " </tr>\n",
357
+ " <tr>\n",
358
+ " <th>4</th>\n",
359
+ " <td>petter mattei love time money visual stun film...</td>\n",
360
+ " <td>1</td>\n",
361
+ " </tr>\n",
362
+ " </tbody>\n",
363
+ "</table>\n",
364
+ "</div>"
365
+ ],
366
+ "text/plain": [
367
+ " review sentiment\n",
368
+ "0 one review mention watch 1 oz episod hook righ... 1\n",
369
+ "1 wonder littl product film techniqu unassum old... 1\n",
370
+ "2 thought wonder way spend time hot summer weeke... 1\n",
371
+ "3 basic famili littl boy jake think zombi closet... 0\n",
372
+ "4 petter mattei love time money visual stun film... 1"
373
+ ]
374
+ },
375
+ "execution_count": 5,
376
+ "metadata": {},
377
+ "output_type": "execute_result"
378
+ }
379
+ ],
380
+ "source": [
381
+ "import re\n",
382
+ "import nltk\n",
383
+ "from nltk.corpus import stopwords\n",
384
+ "from nltk.stem import PorterStemmer\n",
385
+ "\n",
386
+ "\n",
387
+ "def lowercase_text(text):\n",
388
+ " return text.lower()\n",
389
+ "\n",
390
+ "def remove_html(text):\n",
391
+ " return re.sub(r'<[^<]+?>', '', text)\n",
392
+ "\n",
393
+ "def remove_url(text):\n",
394
+ " return re.sub(r'http[s]?://\\S+|www\\.\\S+', '', text)\n",
395
+ "\n",
396
+ "def remove_punctuations(text):\n",
397
+ " tokens_list = '!\"#$%&\\'()*+,-./:;<=>?@[\\\\]^_`{|}~'\n",
398
+ " for char in text:\n",
399
+ " if char in tokens_list:\n",
400
+ " text = text.replace(char, ' ')\n",
401
+ "\n",
402
+ " return text\n",
403
+ "\n",
404
+ "def remove_emojis(text):\n",
405
+ " emojis = re.compile(\"[\"\n",
406
+ " u\"\\U0001F600-\\U0001F64F\"\n",
407
+ " u\"\\U0001F300-\\U0001F5FF\"\n",
408
+ " u\"\\U0001F680-\\U0001F6FF\"\n",
409
+ " u\"\\U0001F1E0-\\U0001F1FF\"\n",
410
+ " u\"\\U00002500-\\U00002BEF\"\n",
411
+ " u\"\\U00002702-\\U000027B0\"\n",
412
+ " u\"\\U00002702-\\U000027B0\"\n",
413
+ " u\"\\U000024C2-\\U0001F251\"\n",
414
+ " u\"\\U0001f926-\\U0001f937\"\n",
415
+ " u\"\\U00010000-\\U0010ffff\"\n",
416
+ " u\"\\u2640-\\u2642\"\n",
417
+ " u\"\\u2600-\\u2B55\"\n",
418
+ " u\"\\u200d\"\n",
419
+ " u\"\\u23cf\"\n",
420
+ " u\"\\u23e9\"\n",
421
+ " u\"\\u231a\"\n",
422
+ " u\"\\ufe0f\"\n",
423
+ " u\"\\u3030\"\n",
424
+ " \"]+\", re.UNICODE)\n",
425
+ "\n",
426
+ " text = re.sub(emojis, '', text)\n",
427
+ " return text\n",
428
+ "\n",
429
+ "def remove_stop_words(text):\n",
430
+ " stop_words = stopwords.words('english')\n",
431
+ " new_text = ''\n",
432
+ " for word in text.split():\n",
433
+ " if word not in stop_words:\n",
434
+ " new_text += ''.join(f'{word} ')\n",
435
+ "\n",
436
+ " return new_text.strip()\n",
437
+ "\n",
438
+ "def stem_words(text):\n",
439
+ " stemmer = PorterStemmer()\n",
440
+ " new_text = ''\n",
441
+ " for word in text.split():\n",
442
+ " new_text += ''.join(f'{stemmer.stem(word)} ')\n",
443
+ "\n",
444
+ " return new_text\n",
445
+ "\n",
446
+ "def preprocess_text(text):\n",
447
+ " text = lowercase_text(text)\n",
448
+ " text = remove_html(text)\n",
449
+ " text = remove_url(text)\n",
450
+ " text = remove_punctuations(text)\n",
451
+ " text = remove_emojis(text)\n",
452
+ " text = remove_stop_words(text)\n",
453
+ " text = stem_words(text)\n",
454
+ "\n",
455
+ " return text\n",
456
+ "\n",
457
+ "nltk.download('stopwords')\n",
458
+ "df_reviews['review'] = df_reviews['review'].apply(preprocess_text)\n",
459
+ "df_reviews.head()"
460
+ ]
461
+ },
462
+ {
463
+ "cell_type": "markdown",
464
+ "metadata": {},
465
+ "source": [
466
+ "### Visualizando balancemento da classes"
467
+ ]
468
+ },
469
+ {
470
+ "cell_type": "code",
471
+ "execution_count": 6,
472
+ "metadata": {
473
+ "colab": {
474
+ "base_uri": "https://localhost:8080/",
475
+ "height": 452
476
+ },
477
+ "id": "Gdi_L0HWfntv",
478
+ "outputId": "bce77594-f662-4b3f-c8eb-27d8a188b4f2"
479
+ },
480
+ "outputs": [
481
+ {
482
+ "data": {
483
+ "image/png": "iVBORw0KGgoAAAANSUhEUgAAAjkAAAGzCAYAAADNKAZOAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjkuMCwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy80BEi2AAAACXBIWXMAAA9hAAAPYQGoP6dpAAA0bElEQVR4nO3deVxVdf7H8TegXHABNAVECRXLfSktJLdKfqKS5mS5NQ46pi3YpDSmpqNmM2NjizppOuUkzYyW1pTlMijhwqhYk8mUlpZ7peCWXFdQ+P7+mAenbuACscS31/PxuI+83/M553zOV+2+vfd7Ll7GGCMAAADLeFd0AwAAAGWBkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsBIhBwAAWImQA6BCDRs2TA0bNqzoNhwHDhyQl5eXkpKSnLFp06bJy8urXM5/++236/bbb3eeb9iwQV5eXnrrrbfK5fw/td8P4Mcg5AClxMvL65oeGzZsqOhWPWzZskXTpk3TqVOnKroVqxw+fFjTpk1TRkZGRbdSyE+5N6A0VanoBgBb/P3vf/d4/re//U0pKSmFxps3b16ebV3Vli1b9NRTT2nYsGEKCgqq6HZ+kiZPnqwJEyYUa5/Dhw/rqaeeUsOGDdWuXbtr3m/t2rXF7K74rtTbK6+8ovz8/DLvASgPhByglPzyl7/0eL5161alpKQUGi8JY4wuXLggf3//H30sFF+VKlVUpUrZ/u/y3Llzqlatmnx9fcv0PFdTtWrVCj0/UJr4uAooR4sWLdKdd96p4OBguVwutWjRQvPnzy9U17BhQ911111as2aNOnToIH9/f/3lL3+RJB08eFB9+/ZV9erVFRwcrLFjx2rNmjVFfhT2wQcfqGfPngoMDFS1atXUrVs3bd682dk+bdo0jRs3TpLUqFEj5yO1AwcOFNn/6NGjVaNGDZ07d67QtsGDBys0NFR5eXmSpHfffVdxcXEKCwuTy+VSZGSknn76aWf75RSsQfnhtRS1VkaSdu3apXvvvVe1a9eWn5+fOnTooPfee++K5yhw6tQpDRs2TIGBgQoKClJ8fHyRH9sVtSYnJSVFnTt3VlBQkGrUqKGmTZvqySefdK7hlltukSQNHz7cmdeC3m+//Xa1atVK27ZtU9euXVWtWjVn3x+uySmQl5enJ598UqGhoapevbr69u2rr776yqOmYcOGGjZsWKF9v3/Mq/VW1Jqcs2fP6vHHH1d4eLhcLpeaNm2q5557TsYYjzovLy+NHj1ay5cvV6tWreRyudSyZUslJycX6gkoD7yTA5Sj+fPnq2XLlurbt6+qVKmiFStW6JFHHlF+fr4SEhI8anfv3q3BgwfrwQcf1MiRI9W0aVOdPXtWd955p44cOaLHHntMoaGhWrJkidavX1/oXOvWrVOvXr3Uvn17TZ06Vd7e3k7I+ve//61bb71V99xzj7744gu9/vrrmjVrlurUqSNJqlu3bpH9Dxw4UPPmzdOqVat03333OePnzp3TihUrNGzYMPn4+EiSkpKSVKNGDSUmJqpGjRpat26dpkyZIrfbrWeffbZU5nPnzp3q1KmT6tevrwkTJqh69epatmyZ+vXrp3/+85/6xS9+cdl9jTG6++67tWnTJj300ENq3ry53nnnHcXHx1/Tee+66y61adNG06dPl8vl0p49e5wA2bx5c02fPl1TpkzRqFGj1KVLF0nSbbfd5hzjxIkT6tWrlwYNGqRf/vKXCgkJueI5//CHP8jLy0vjx4/X0aNHNXv2bMXExCgjI6NY7/BdS2/fZ4xR3759tX79eo0YMULt2rXTmjVrNG7cOH3zzTeaNWuWR/2mTZv09ttv65FHHlHNmjX15z//Wf3799ehQ4d03XXXXXOfQKkwAMpEQkKC+eFfsXPnzhWqi42NNY0bN/YYi4iIMJJMcnKyx/jzzz9vJJnly5c7Y+fPnzfNmjUzksz69euNMcbk5+ebG264wcTGxpr8/HyP8zdq1Mj83//9nzP27LPPGklm//79V72m/Px8U79+fdO/f3+P8WXLlhlJJi0t7YrX+uCDD5pq1aqZCxcuOGPx8fEmIiLCeb5+/XqPaymwf/9+I8ksWrTIGevevbtp3bq1x/Hy8/PNbbfdZm644YYrXsvy5cuNJDNz5kxn7NKlS6ZLly6FzjN16lSP38tZs2YZSebYsWOXPf5//vOfQscp0K1bNyPJLFiwoMht3bp1c54XzEf9+vWN2+12xgvmfM6cOc5YRESEiY+Pv+oxr9TbD38/Cubp97//vUfdvffea7y8vMyePXucMUnG19fXY+y///2vkWRefPHFQucCyhofVwHl6Pv/4s7Oztbx48fVrVs37du3T9nZ2R61jRo1UmxsrMdYcnKy6tevr759+zpjfn5+GjlypEddRkaGvvzySw0ZMkQnTpzQ8ePHdfz4cZ09e1bdu3dXWlpaiRaXenl56b777tPq1at15swZZ3zp0qWqX7++OnfuXOS1nj59WsePH1eXLl107tw57dq1q9jn/qGTJ09q3bp1GjBggHP848eP68SJE4qNjdWXX36pb7755rL7r169WlWqVNHDDz/sjPn4+OjRRx+96rkLFmi/++67JV6k63K5NHz48Guu/9WvfqWaNWs6z++9917Vq1dPq1evLtH5r9Xq1avl4+Oj3/zmNx7jjz/+uIwx+te//uUxHhMTo8jISOd5mzZtFBAQoH379pVpn0BRCDlAOdq8ebNiYmJUvXp1BQUFqW7dus5ajKJCzg8dPHhQkZGRhdaHNGnSxOP5l19+KUmKj49X3bp1PR4LFy5UTk5OofNdq4EDB+r8+fPOupczZ85o9erVuu+++zz62rlzp37xi18oMDBQAQEBqlu3rrMIu6Tn/r49e/bIGKPf/e53ha5x6tSpkqSjR49edv+DBw+qXr16qlGjhsd406ZNr3rugQMHqlOnTnrggQcUEhKiQYMGadmyZcUKPPXr1y/WIuMbbrjB47mXl5eaNGly2fVTpeXgwYMKCwvzCFjSd3cJHjx40GP8+uuvL3SMWrVq6dtvvy27JoHLYE0OUE727t2r7t27q1mzZnrhhRcUHh4uX19frV69WrNmzSr0Avlj7qQqONazzz572duXf/jifq06duyohg0batmyZRoyZIhWrFih8+fPa+DAgU7NqVOn1K1bNwUEBGj69OmKjIyUn5+fPv74Y40fP/6KYeByX7r3wwXLBcf47W9/W+gdrwI/DH+lxd/fX2lpaVq/fr1WrVql5ORkLV26VHfeeafWrl3rrEu62jFK25Xm7lp6Kg2XO4/5wSJloDwQcoBysmLFCuXk5Oi9997z+NduUYuGLyciIkKfffaZjDEeL2h79uzxqCv4uCAgIEAxMTFXPGZJvsl3wIABmjNnjtxut5YuXaqGDRuqY8eOzvYNGzboxIkTevvtt9W1a1dnfP/+/Vc9dq1atSSp0F1OP3zHoHHjxpL+d8vz1a6xKBEREUpNTdWZM2c8At/u3buvaX9vb291795d3bt31wsvvKA//vGPmjRpktavX6+YmJhS/4bkgnfnChhjtGfPHrVp08YZq1WrVpF3hx08eNCZL6l4v+cRERF6//33dfr0aY93cwo+coyIiLjmYwHljY+rgHJS8C/c7/+LNjs7W4sWLbrmY8TGxuqbb77xuEX6woULeuWVVzzq2rdvr8jISD333HMea2cKHDt2zPl19erVJRUOFVcycOBA5eTk6LXXXlNycrIGDBjgsb2oa83NzdVLL7101WNHRETIx8dHaWlpHuM/3Dc4OFi33367/vKXv+jIkSOFjvP9ayxK7969denSJY9b+PPy8vTiiy9etceTJ08WGit4xywnJ0dSyeb1Sv72t7/p9OnTzvO33npLR44cUa9evZyxyMhIbd26Vbm5uc7YypUrC91qXpzeevfurby8PM2dO9djfNasWfLy8vI4P/BTwzs5QDnp0aOHfH191adPHz344IM6c+aMXnnlFQUHBxf5Il2UBx98UHPnztXgwYP12GOPqV69elq8eLH8/PwkffcvdG9vby1cuFC9evVSy5YtNXz4cNWvX1/ffPON1q9fr4CAAK1YsULS/wKRJE2aNEmDBg1S1apV1adPH+eFsCg333yzmjRpokmTJiknJ8fjoyrpf7cj16pVS/Hx8frNb34jLy8v/f3vf7+mjywCAwN133336cUXX5SXl5ciIyO1cuXKItfXzJs3T507d1br1q01cuRINW7cWFlZWUpPT9fXX3+t//73v5c9T58+fdSpUydNmDBBBw4cUIsWLfT2229f03qh6dOnKy0tTXFxcYqIiNDRo0f10ksvqUGDBs7i68jISAUFBWnBggWqWbOmqlevrqioqCLXWl2L2rVrq3Pnzho+fLiysrI0e/ZsNWnSxGPR+QMPPKC33npLPXv21IABA7R371794x//8FgIXNze+vTpozvuuEOTJk3SgQMH1LZtW61du1bvvvuuxowZU+jYwE9Kxd3YBditqFvI33vvPdOmTRvj5+dnGjZsaP70pz+ZV199tdAt3BERESYuLq7I4+7bt8/ExcUZf39/U7duXfP444+bf/7zn0aS2bp1q0ft9u3bzT333GOuu+4643K5TEREhBkwYIBJTU31qHv66adN/fr1jbe39zXfTj5p0iQjyTRp0qTI7Zs3bzYdO3Y0/v7+JiwszDzxxBNmzZo1hW4P/+Ety8YYc+zYMdO/f39TrVo1U6tWLfPggw+aHTt2FHnb8969e82vfvUrExoaaqpWrWrq169v7rrrLvPWW29d9RpOnDhhhg4dagICAkxgYKAZOnSo2b59+1VvIU9NTTV33323CQsLM76+viYsLMwMHjzYfPHFFx7Hf/fdd02LFi1MlSpVPI7ZrVs307JlyyJ7utwt5K+//rqZOHGiCQ4ONv7+/iYuLs4cPHiw0P7PP/+8qV+/vnG5XKZTp07mo48+KnTMK/VW1O/H6dOnzdixY01YWJipWrWqueGGG8yzzz7r8fUExvzvFvKEhIRCPV3u1nagrHkZw2owoLKbPXu2xo4dq6+//lr169ev6HYA4CeBkANUMufPn/e4M+fChQu66aablJeXpy+++KICOwOAnxbW5ACVzD333KPrr79e7dq1U3Z2tv7xj39o165dWrx4cUW3BgA/KYQcoJKJjY3VwoULtXjxYuXl5alFixZ64403Ci3+BYCfOz6uAgAAVuJ7cgAAgJUIOQAAwEo/6zU5+fn5Onz4sGrWrFnqX8EOAADKhjFGp0+fVlhYmLy9L/9+zc865Bw+fFjh4eEV3QYAACiBr776Sg0aNLjs9p91yCn4YXNfffWVAgICKrgbAABwLdxut8LDwz1+aGxRftYhp+AjqoCAAEIOAACVzNWWmrDwGAAAWImQAwAArETIAQAAViLkAAAAKxFyAACAlQg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsVKyQM2PGDN1yyy2qWbOmgoOD1a9fP+3evduj5vbbb5eXl5fH46GHHvKoOXTokOLi4lStWjUFBwdr3LhxunTpkkfNhg0bdPPNN8vlcqlJkyZKSkoq1M+8efPUsGFD+fn5KSoqSh9++GFxLgcAAFisWCFn48aNSkhI0NatW5WSkqKLFy+qR48eOnv2rEfdyJEjdeTIEecxc+ZMZ1teXp7i4uKUm5urLVu26LXXXlNSUpKmTJni1Ozfv19xcXG64447lJGRoTFjxuiBBx7QmjVrnJqlS5cqMTFRU6dO1ccff6y2bdsqNjZWR48eLelcAAAAi3gZY0xJdz527JiCg4O1ceNGde3aVdL/3slp166dZs+eXeQ+//rXv3TXXXfp8OHDCgkJkSQtWLBA48eP17Fjx+Tr66vx48dr1apV2rFjh7PfoEGDdOrUKSUnJ0uSoqKidMstt2ju3LmSpPz8fIWHh+vRRx/VhAkTrql/t9utwMBAZWdn8wM6AQCoJK719ftHrcnJzs6WJNWuXdtjfPHixapTp45atWqliRMn6ty5c8629PR0tW7d2gk4khQbGyu3262dO3c6NTExMR7HjI2NVXp6uiQpNzdX27Zt86jx9vZWTEyMU1OUnJwcud1ujwcAALBTlZLumJ+frzFjxqhTp05q1aqVMz5kyBBFREQoLCxMn3zyicaPH6/du3fr7bffliRlZmZ6BBxJzvPMzMwr1rjdbp0/f17ffvut8vLyiqzZtWvXZXueMWOGnnrqqZJecrE0nLCqXM5Tmg48E1fRLQAALoPXleIrcchJSEjQjh07tGnTJo/xUaNGOb9u3bq16tWrp+7du2vv3r2KjIwseaelYOLEiUpMTHSeu91uhYeHV2BHAACgrJQo5IwePVorV65UWlqaGjRocMXaqKgoSdKePXsUGRmp0NDQQndBZWVlSZJCQ0Od/xaMfb8mICBA/v7+8vHxkY+PT5E1BccoisvlksvluraLBAAAlVqx1uQYYzR69Gi98847WrdunRo1anTVfTIyMiRJ9erVkyRFR0fr008/9bgLKiUlRQEBAWrRooVTk5qa6nGclJQURUdHS5J8fX3Vvn17j5r8/HylpqY6NQAA4OetWO/kJCQkaMmSJXr33XdVs2ZNZw1NYGCg/P39tXfvXi1ZskS9e/fWddddp08++URjx45V165d1aZNG0lSjx491KJFCw0dOlQzZ85UZmamJk+erISEBOddloceekhz587VE088oV//+tdat26dli1bplWrvvs8MjExUfHx8erQoYNuvfVWzZ49W2fPntXw4cNLa24AAEAlVqyQM3/+fEn/u038+xYtWqRhw4bJ19dX77//vhM4wsPD1b9/f02ePNmp9fHx0cqVK/Xwww8rOjpa1atXV3x8vKZPn+7UNGrUSKtWrdLYsWM1Z84cNWjQQAsXLlRsbKxTM3DgQB07dkxTpkxRZmam2rVrp+Tk5EKLkQEAwM/Tj/qenMquLL8nh1XwAIDSxOvKd8rle3IAAAB+qgg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsRMgBAABWIuQAAAArEXIAAICVCDkAAMBKhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsBIhBwAAWImQAwAArETIAQAAViLkAAAAKxFyAACAlQg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsRMgBAABWIuQAAAArEXIAAICVCDkAAMBKhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsBIhBwAAWImQAwAArETIAQAAViLkAAAAKxFyAACAlQg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsRMgBAABWIuQAAAArEXIAAICVCDkAAMBKhBwAAGAlQg4AALBSsULOjBkzdMstt6hmzZoKDg5Wv379tHv3bo+aCxcuKCEhQdddd51q1Kih/v37Kysry6Pm0KFDiouLU7Vq1RQcHKxx48bp0qVLHjUbNmzQzTffLJfLpSZNmigpKalQP/PmzVPDhg3l5+enqKgoffjhh8W5HAAAYLFihZyNGzcqISFBW7duVUpKii5evKgePXro7NmzTs3YsWO1YsUKvfnmm9q4caMOHz6se+65x9mel5enuLg45ebmasuWLXrttdeUlJSkKVOmODX79+9XXFyc7rjjDmVkZGjMmDF64IEHtGbNGqdm6dKlSkxM1NSpU/Xxxx+rbdu2io2N1dGjR3/MfAAAAEt4GWNMSXc+duyYgoODtXHjRnXt2lXZ2dmqW7eulixZonvvvVeStGvXLjVv3lzp6enq2LGj/vWvf+muu+7S4cOHFRISIklasGCBxo8fr2PHjsnX11fjx4/XqlWrtGPHDudcgwYN0qlTp5ScnCxJioqK0i233KK5c+dKkvLz8xUeHq5HH31UEyZMuKb+3W63AgMDlZ2drYCAgJJOQ5EaTlhVqscrDweeiavoFgAAl8Hryneu9fX7R63Jyc7OliTVrl1bkrRt2zZdvHhRMTExTk2zZs10/fXXKz09XZKUnp6u1q1bOwFHkmJjY+V2u7Vz506n5vvHKKgpOEZubq62bdvmUePt7a2YmBinpig5OTlyu90eDwAAYKcSh5z8/HyNGTNGnTp1UqtWrSRJmZmZ8vX1VVBQkEdtSEiIMjMznZrvB5yC7QXbrlTjdrt1/vx5HT9+XHl5eUXWFByjKDNmzFBgYKDzCA8PL/6FAwCASqHEISchIUE7duzQG2+8UZr9lKmJEycqOzvbeXz11VcV3RIAACgjVUqy0+jRo7Vy5UqlpaWpQYMGznhoaKhyc3N16tQpj3dzsrKyFBoa6tT88C6ogruvvl/zwzuysrKyFBAQIH9/f/n4+MjHx6fImoJjFMXlcsnlchX/ggEAQKVTrHdyjDEaPXq03nnnHa1bt06NGjXy2N6+fXtVrVpVqampztju3bt16NAhRUdHS5Kio6P16aefetwFlZKSooCAALVo0cKp+f4xCmoKjuHr66v27dt71OTn5ys1NdWpAQAAP2/FeicnISFBS5Ys0bvvvquaNWs6618CAwPl7++vwMBAjRgxQomJiapdu7YCAgL06KOPKjo6Wh07dpQk9ejRQy1atNDQoUM1c+ZMZWZmavLkyUpISHDeZXnooYc0d+5cPfHEE/r1r3+tdevWadmyZVq16ruV5YmJiYqPj1eHDh106623avbs2Tp79qyGDx9eWnMDAAAqsWKFnPnz50uSbr/9do/xRYsWadiwYZKkWbNmydvbW/3791dOTo5iY2P10ksvObU+Pj5auXKlHn74YUVHR6t69eqKj4/X9OnTnZpGjRpp1apVGjt2rObMmaMGDRpo4cKFio2NdWoGDhyoY8eOacqUKcrMzFS7du2UnJxcaDEyAAD4efpR35NT2fE9OZ74nhwA+OnideU75fI9OQAAAD9VhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsBIhBwAAWImQAwAArETIAQAAViLkAAAAKxFyAACAlQg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsRMgBAABWIuQAAAArEXIAAICVCDkAAMBKhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsBIhBwAAWImQAwAArETIAQAAViLkAAAAKxFyAACAlQg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsRMgBAABWIuQAAAArEXIAAICVCDkAAMBKhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsFKxQ05aWpr69OmjsLAweXl5afny5R7bhw0bJi8vL49Hz549PWpOnjyp+++/XwEBAQoKCtKIESN05swZj5pPPvlEXbp0kZ+fn8LDwzVz5sxCvbz55ptq1qyZ/Pz81Lp1a61evbq4lwMAACxV7JBz9uxZtW3bVvPmzbtsTc+ePXXkyBHn8frrr3tsv//++7Vz506lpKRo5cqVSktL06hRo5ztbrdbPXr0UEREhLZt26Znn31W06ZN08svv+zUbNmyRYMHD9aIESO0fft29evXT/369dOOHTuKe0kAAMBCVYq7Q69evdSrV68r1rhcLoWGhha57fPPP1dycrL+85//qEOHDpKkF198Ub1799Zzzz2nsLAwLV68WLm5uXr11Vfl6+urli1bKiMjQy+88IIThubMmaOePXtq3LhxkqSnn35aKSkpmjt3rhYsWFDcywIAAJYpkzU5GzZsUHBwsJo2baqHH35YJ06ccLalp6crKCjICTiSFBMTI29vb33wwQdOTdeuXeXr6+vUxMbGavfu3fr222+dmpiYGI/zxsbGKj09/bJ95eTkyO12ezwAAICdSj3k9OzZU3/729+UmpqqP/3pT9q4caN69eqlvLw8SVJmZqaCg4M99qlSpYpq166tzMxMpyYkJMSjpuD51WoKthdlxowZCgwMdB7h4eE/7mIBAMBPVrE/rrqaQYMGOb9u3bq12rRpo8jISG3YsEHdu3cv7dMVy8SJE5WYmOg8d7vdBB0AACxV5reQN27cWHXq1NGePXskSaGhoTp69KhHzaVLl3Ty5ElnHU9oaKiysrI8agqeX63mcmuBpP+tFQoICPB4AAAAO5V5yPn666914sQJ1atXT5IUHR2tU6dOadu2bU7NunXrlJ+fr6ioKKcmLS1NFy9edGpSUlLUtGlT1apVy6lJTU31OFdKSoqio6PL+pIAAEAlUOyQc+bMGWVkZCgjI0OStH//fmVkZOjQoUM6c+aMxo0bp61bt+rAgQNKTU3V3XffrSZNmig2NlaS1Lx5c/Xs2VMjR47Uhx9+qM2bN2v06NEaNGiQwsLCJElDhgyRr6+vRowYoZ07d2rp0qWaM2eOx0dNjz32mJKTk/X8889r165dmjZtmj766CONHj26FKYFAABUdsUOOR999JFuuukm3XTTTZKkxMRE3XTTTZoyZYp8fHz0ySefqG/fvrrxxhs1YsQItW/fXv/+97/lcrmcYyxevFjNmjVT9+7d1bt3b3Xu3NnjO3ACAwO1du1a7d+/X+3bt9fjjz+uKVOmeHyXzm233aYlS5bo5ZdfVtu2bfXWW29p+fLlatWq1Y+ZDwAAYAkvY4yp6CYqitvtVmBgoLKzs0t9fU7DCatK9Xjl4cAzcRXdAgDgMnhd+c61vn7zs6sAAICVCDkAAMBKhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsBIhBwAAWImQAwAArETIAQAAViLkAAAAKxFyAACAlQg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsRMgBAABWIuQAAAArEXIAAICVCDkAAMBKhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsBIhBwAAWImQAwAArETIAQAAViLkAAAAKxFyAACAlQg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsRMgBAABWIuQAAAArEXIAAICVCDkAAMBKhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsFKxQ05aWpr69OmjsLAweXl5afny5R7bjTGaMmWK6tWrJ39/f8XExOjLL7/0qDl58qTuv/9+BQQEKCgoSCNGjNCZM2c8aj755BN16dJFfn5+Cg8P18yZMwv18uabb6pZs2by8/NT69attXr16uJeDgAAsFSxQ87Zs2fVtm1bzZs3r8jtM2fO1J///GctWLBAH3zwgapXr67Y2FhduHDBqbn//vu1c+dOpaSkaOXKlUpLS9OoUaOc7W63Wz169FBERIS2bdumZ599VtOmTdPLL7/s1GzZskWDBw/WiBEjtH37dvXr10/9+vXTjh07intJAADAQl7GGFPinb289M4776hfv36S/vcuTlhYmB5//HH99re/lSRlZ2crJCRESUlJGjRokD7//HO1aNFC//nPf9ShQwdJUnJysnr37q2vv/5aYWFhmj9/viZNmqTMzEz5+vpKkiZMmKDly5dr165dkqSBAwfq7NmzWrlypdNPx44d1a5dOy1YsKDIfnNycpSTk+M8d7vdCg8PV3Z2tgICAko6DUVqOGFVqR6vPBx4Jq6iWwAAXAavK99xu90KDAy86ut3qa7J2b9/vzIzMxUTE+OMBQYGKioqSunp6ZKk9PR0BQUFOQFHkmJiYuTt7a0PPvjAqenatasTcCQpNjZWu3fv1rfffuvUfP88BTUF5ynKjBkzFBgY6DzCw8N//EUDAICfpFINOZmZmZKkkJAQj/GQkBBnW2ZmpoKDgz22V6lSRbVr1/aoKeoY3z/H5WoKthdl4sSJys7Odh5fffVVcS8RAABUElUquoHy5HK55HK5KroNAABQDkr1nZzQ0FBJUlZWlsd4VlaWsy00NFRHjx712H7p0iWdPHnSo6aoY3z/HJerKdgOAAB+3ko15DRq1EihoaFKTU11xtxutz744ANFR0dLkqKjo3Xq1Clt27bNqVm3bp3y8/MVFRXl1KSlpenixYtOTUpKipo2bapatWo5Nd8/T0FNwXkAAMDPW7FDzpkzZ5SRkaGMjAxJ/1tsnJGRoUOHDsnLy0tjxozR73//e7333nv69NNP9atf/UphYWHOHVjNmzdXz549NXLkSH344YfavHmzRo8erUGDBiksLEySNGTIEPn6+mrEiBHauXOnli5dqjlz5igxMdHp47HHHlNycrKef/557dq1S9OmTdNHH32k0aNH//hZAQAAlV6x1+R89NFHuuOOO5znBcEjPj5eSUlJeuKJJ3T27FmNGjVKp06dUufOnZWcnCw/Pz9nn8WLF2v06NHq3r27vL291b9/f/35z392tgcGBmrt2rVKSEhQ+/btVadOHU2ZMsXju3Ruu+02LVmyRJMnT9aTTz6pG264QcuXL1erVq1KNBEAAMAuP+p7ciq7a73PviT4PgMAQGnideU7FfI9OQAAAD8VhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsBIhBwAAWImQAwAArETIAQAAViLkAAAAKxFyAACAlQg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsRMgBAABWIuQAAAArEXIAAICVCDkAAMBKhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsBIhBwAAWImQAwAArETIAQAAViLkAAAAKxFyAACAlQg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsRMgBAABWIuQAAAArEXIAAICVCDkAAMBKhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsFKph5xp06bJy8vL49GsWTNn+4ULF5SQkKDrrrtONWrUUP/+/ZWVleVxjEOHDikuLk7VqlVTcHCwxo0bp0uXLnnUbNiwQTfffLNcLpeaNGmipKSk0r4UAABQiZXJOzktW7bUkSNHnMemTZucbWPHjtWKFSv05ptvauPGjTp8+LDuueceZ3teXp7i4uKUm5urLVu26LXXXlNSUpKmTJni1Ozfv19xcXG64447lJGRoTFjxuiBBx7QmjVryuJyAABAJVSlTA5apYpCQ0MLjWdnZ+uvf/2rlixZojvvvFOStGjRIjVv3lxbt25Vx44dtXbtWn322Wd6//33FRISonbt2unpp5/W+PHjNW3aNPn6+mrBggVq1KiRnn/+eUlS8+bNtWnTJs2aNUuxsbFlcUkAAKCSKZN3cr788kuFhYWpcePGuv/++3Xo0CFJ0rZt23Tx4kXFxMQ4tc2aNdP111+v9PR0SVJ6erpat26tkJAQpyY2NlZut1s7d+50ar5/jIKagmNcTk5Ojtxut8cDAADYqdRDTlRUlJKSkpScnKz58+dr//796tKli06fPq3MzEz5+voqKCjIY5+QkBBlZmZKkjIzMz0CTsH2gm1XqnG73Tp//vxle5sxY4YCAwOdR3h4+I+9XAAA8BNV6h9X9erVy/l1mzZtFBUVpYiICC1btkz+/v6lfbpimThxohITE53nbreboAMAgKXK/BbyoKAg3XjjjdqzZ49CQ0OVm5urU6dOedRkZWU5a3hCQ0ML3W1V8PxqNQEBAVcMUi6XSwEBAR4PAABgpzIPOWfOnNHevXtVr149tW/fXlWrVlVqaqqzfffu3Tp06JCio6MlSdHR0fr000919OhRpyYlJUUBAQFq0aKFU/P9YxTUFBwDAACg1EPOb3/7W23cuFEHDhzQli1b9Itf/EI+Pj4aPHiwAgMDNWLECCUmJmr9+vXatm2bhg8frujoaHXs2FGS1KNHD7Vo0UJDhw7Vf//7X61Zs0aTJ09WQkKCXC6XJOmhhx7Svn379MQTT2jXrl166aWXtGzZMo0dO7a0LwcAAFRSpb4m5+uvv9bgwYN14sQJ1a1bV507d9bWrVtVt25dSdKsWbPk7e2t/v37KycnR7GxsXrppZec/X18fLRy5Uo9/PDDio6OVvXq1RUfH6/p06c7NY0aNdKqVas0duxYzZkzRw0aNNDChQu5fRwAADi8jDGmopuoKG63W4GBgcrOzi719TkNJ6wq1eOVhwPPxFV0CwCAy+B15TvX+vrNz64CAABWIuQAAAArEXIAAICVCDkAAMBKhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsBIhBwAAWImQAwAArETIAQAAViLkAAAAKxFyAACAlQg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsRMgBAABWIuQAAAArEXIAAICVCDkAAMBKhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUIOQAAwEqEHAAAYCVCDgAAsBIhBwAAWImQAwAArETIAQAAViLkAAAAKxFyAACAlQg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsRMgBAABWIuQAAAArEXIAAICVCDkAAMBKhBwAAGAlQg4AALASIQcAAFiJkAMAAKxEyAEAAFYi5AAAACsRcgAAgJUqfciZN2+eGjZsKD8/P0VFRenDDz+s6JYAAMBPQKUOOUuXLlViYqKmTp2qjz/+WG3btlVsbKyOHj1a0a0BAIAKVqlDzgsvvKCRI0dq+PDhatGihRYsWKBq1arp1VdfrejWAABABatS0Q2UVG5urrZt26aJEyc6Y97e3oqJiVF6enqR++Tk5CgnJ8d5np2dLUlyu92l3l9+zrlSP2ZZK4t5AACUDl5XCh/XGHPFukobco4fP668vDyFhIR4jIeEhGjXrl1F7jNjxgw99dRThcbDw8PLpMfKJnB2RXcAALBJWb+unD59WoGBgZfdXmlDTklMnDhRiYmJzvP8/HydPHlS1113nby8vErtPG63W+Hh4frqq68UEBBQaseFJ+a5/DDX5YN5Lh/Mc/koy3k2xuj06dMKCwu7Yl2lDTl16tSRj4+PsrKyPMazsrIUGhpa5D4ul0sul8tjLCgoqKxaVEBAAH+BygHzXH6Y6/LBPJcP5rl8lNU8X+kdnAKVduGxr6+v2rdvr9TUVGcsPz9fqampio6OrsDOAADAT0GlfSdHkhITExUfH68OHTro1ltv1ezZs3X27FkNHz68olsDAAAVrFKHnIEDB+rYsWOaMmWKMjMz1a5dOyUnJxdajFzeXC6Xpk6dWuijMZQu5rn8MNflg3kuH8xz+fgpzLOXudr9VwAAAJVQpV2TAwAAcCWEHAAAYCVCDgAAsBIhBwAAWImQAwAArETIKaF58+apYcOG8vPzU1RUlD788MMr1r/55ptq1qyZ/Pz81Lp1a61evbqcOq3cijPPr7zyirp06aJatWqpVq1aiomJuervC/6nuH+eC7zxxhvy8vJSv379yrZBixR3rk+dOqWEhATVq1dPLpdLN954I///uAbFnefZs2eradOm8vf3V3h4uMaOHasLFy6UU7eVU1pamvr06aOwsDB5eXlp+fLlV91nw4YNuvnmm+VyudSkSRMlJSWVbZMGxfbGG28YX19f8+qrr5qdO3eakSNHmqCgIJOVlVVk/ebNm42Pj4+ZOXOm+eyzz8zkyZNN1apVzaefflrOnVcuxZ3nIUOGmHnz5pnt27ebzz//3AwbNswEBgaar7/+upw7r1yKO88F9u/fb+rXr2+6dOli7r777vJptpIr7lzn5OSYDh06mN69e5tNmzaZ/fv3mw0bNpiMjIxy7rxyKe48L1682LhcLrN48WKzf/9+s2bNGlOvXj0zduzYcu68clm9erWZNGmSefvtt40k884771yxft++faZatWomMTHRfPbZZ+bFF180Pj4+Jjk5ucx6JOSUwK233moSEhKc53l5eSYsLMzMmDGjyPoBAwaYuLg4j7GoqCjz4IMPlmmflV1x5/mHLl26ZGrWrGlee+21smrRCiWZ50uXLpnbbrvNLFy40MTHxxNyrlFx53r+/PmmcePGJjc3t7xatEJx5zkhIcHceeedHmOJiYmmU6dOZdqnTa4l5DzxxBOmZcuWHmMDBw40sbGxZdYXH1cVU25urrZt26aYmBhnzNvbWzExMUpPTy9yn/T0dI96SYqNjb1sPUo2zz907tw5Xbx4UbVr1y6rNiu9ks7z9OnTFRwcrBEjRpRHm1YoyVy/9957io6OVkJCgkJCQtSqVSv98Y9/VF5eXnm1XemUZJ5vu+02bdu2zflIa9++fVq9erV69+5dLj3/XFTEa2Gl/rEOFeH48ePKy8sr9KMjQkJCtGvXriL3yczMLLI+MzOzzPqs7Eoyzz80fvx4hYWFFfpLhe+UZJ43bdqkv/71r8rIyCiHDu1Rkrnet2+f1q1bp/vvv1+rV6/Wnj179Mgjj+jixYuaOnVqebRd6ZRknocMGaLjx4+rc+fOMsbo0qVLeuihh/Tkk0+WR8s/G5d7LXS73Tp//rz8/f1L/Zy8kwMrPfPMM3rjjTf0zjvvyM/Pr6Lbscbp06c1dOhQvfLKK6pTp05Ft2O9/Px8BQcH6+WXX1b79u01cOBATZo0SQsWLKjo1qyyYcMG/fGPf9RLL72kjz/+WG+//bZWrVqlp59+uqJbw4/EOznFVKdOHfn4+CgrK8tjPCsrS6GhoUXuExoaWqx6lGyeCzz33HN65pln9P7776tNmzZl2WalV9x53rt3rw4cOKA+ffo4Y/n5+ZKkKlWqaPfu3YqMjCzbpiupkvyZrlevnqpWrSofHx9nrHnz5srMzFRubq58fX3LtOfKqCTz/Lvf/U5Dhw7VAw88IElq3bq1zp49q1GjRmnSpEny9ub9gNJwudfCgICAMnkXR+KdnGLz9fVV+/btlZqa6ozl5+crNTVV0dHRRe4THR3tUS9JKSkpl61HyeZZkmbOnKmnn35aycnJ6tChQ3m0WqkVd56bNWumTz/9VBkZGc6jb9++uuOOO5SRkaHw8PDybL9SKcmf6U6dOmnPnj1OkJSkL774QvXq1SPgXEZJ5vncuXOFgkxBsDT8DOtSUyGvhWW2pNlib7zxhnG5XCYpKcl89tlnZtSoUSYoKMhkZmYaY4wZOnSomTBhglO/efNmU6VKFfPcc8+Zzz//3EydOpVbyK9Bcef5mWeeMb6+vuatt94yR44ccR6nT5+uqEuoFIo7zz/E3VXXrrhzfejQIVOzZk0zevRos3v3brNy5UoTHBxsfv/731fUJVQKxZ3nqVOnmpo1a5rXX3/d7Nu3z6xdu9ZERkaaAQMGVNQlVAqnT58227dvN9u3bzeSzAsvvGC2b99uDh48aIwxZsKECWbo0KFOfcEt5OPGjTOff/65mTdvHreQ/1S9+OKL5vrrrze+vr7m1ltvNVu3bnW2devWzcTHx3vUL1u2zNx4443G19fXtGzZ0qxataqcO66cijPPERERRlKhx9SpU8u/8UqmuH+ev4+QUzzFnestW7aYqKgo43K5TOPGjc0f/vAHc+nSpXLuuvIpzjxfvHjRTJs2zURGRho/Pz8THh5uHnnkEfPtt9+Wf+OVyPr164v8f27B3MbHx5tu3boV2qddu3bG19fXNG7c2CxatKhMe/QyhvfiAACAfViTAwAArETIAQAAViLkAAAAKxFyAACAlQg5AADASoQcAABgJUIOAACwEiEHAABYiZADAACsRMgBAABWIuQAAAAr/T9WUkYIDg5fygAAAABJRU5ErkJggg==",
484
+ "text/plain": [
485
+ "<Figure size 640x480 with 1 Axes>"
486
+ ]
487
+ },
488
+ "metadata": {},
489
+ "output_type": "display_data"
490
+ }
491
+ ],
492
+ "source": [
493
+ "plt.title('Target value distribution')\n",
494
+ "plt.hist(df_reviews['sentiment'])\n",
495
+ "plt.show()"
496
+ ]
497
+ },
498
+ {
499
+ "cell_type": "markdown",
500
+ "metadata": {},
501
+ "source": [
502
+ "# Modelo BERT"
503
+ ]
504
+ },
505
+ {
506
+ "cell_type": "markdown",
507
+ "metadata": {
508
+ "id": "EDkjlPDakskM"
509
+ },
510
+ "source": [
511
+ "## Instalando Bibliotecas"
512
+ ]
513
+ },
514
+ {
515
+ "cell_type": "code",
516
+ "execution_count": 4,
517
+ "metadata": {
518
+ "colab": {
519
+ "base_uri": "https://localhost:8080/"
520
+ },
521
+ "id": "lk7m_1xvmWvz",
522
+ "outputId": "ce842053-b261-4768-d9d7-fe9c65c9f6aa"
523
+ },
524
+ "outputs": [],
525
+ "source": [
526
+ "#pip install transformers\n",
527
+ "#pip install accelerate -U\n",
528
+ "#pip install transformers[torch]\n",
529
+ "#pip install datasets evaluate"
530
+ ]
531
+ },
532
+ {
533
+ "cell_type": "markdown",
534
+ "metadata": {},
535
+ "source": [
536
+ "## Carregando o modelo treinado e tokenizador"
537
+ ]
538
+ },
539
+ {
540
+ "cell_type": "code",
541
+ "execution_count": 10,
542
+ "metadata": {
543
+ "colab": {
544
+ "base_uri": "https://localhost:8080/"
545
+ },
546
+ "id": "GlyrkK52zMcc",
547
+ "outputId": "a938653b-92c3-4b4e-802c-eacc3f1b6ecf"
548
+ },
549
+ "outputs": [
550
+ {
551
+ "name": "stderr",
552
+ "output_type": "stream",
553
+ "text": [
554
+ "c:\\Users\\andre\\1JUPYTER\\dt_labs\\.venv\\Lib\\site-packages\\tqdm\\auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
555
+ " from .autonotebook import tqdm as notebook_tqdm\n"
556
+ ]
557
+ }
558
+ ],
559
+ "source": [
560
+ "from transformers import AutoTokenizer\n",
561
+ "from transformers import BertForSequenceClassification\n",
562
+ "\n",
563
+ "pre_trained_base = \"bert-base-uncased\"\n",
564
+ "tokenizer = AutoTokenizer.from_pretrained(pre_trained_base)\n",
565
+ "model = BertForSequenceClassification.from_pretrained(pre_trained_base, num_labels = 2, output_attentions=False, output_hidden_states=False)"
566
+ ]
567
+ },
568
+ {
569
+ "cell_type": "markdown",
570
+ "metadata": {},
571
+ "source": [
572
+ "### Tokenização das Sentenças e CÑlculo do Tamanho dos Tokens"
573
+ ]
574
+ },
575
+ {
576
+ "cell_type": "code",
577
+ "execution_count": 13,
578
+ "metadata": {
579
+ "id": "LKEjDZCHpk4e"
580
+ },
581
+ "outputs": [],
582
+ "source": [
583
+ "token_lens = []\n",
584
+ "\n",
585
+ "for sentence in df_reviews['review']:\n",
586
+ " tokens = tokenizer.encode(sentence, max_length=200, truncation=True)\n",
587
+ " token_lens.append(len(tokens))"
588
+ ]
589
+ },
590
+ {
591
+ "cell_type": "markdown",
592
+ "metadata": {},
593
+ "source": [
594
+ "### Divisão dos Dados em Conjunto de Treinamento e Validação:"
595
+ ]
596
+ },
597
+ {
598
+ "cell_type": "code",
599
+ "execution_count": 15,
600
+ "metadata": {
601
+ "id": "H7PfXaVVp2uQ"
602
+ },
603
+ "outputs": [],
604
+ "source": [
605
+ "SEED=42\n",
606
+ "MAX_LEN = 200\n",
607
+ "from sklearn.model_selection import train_test_split\n",
608
+ "df_train, df_val = train_test_split(df_reviews, test_size=0.2, random_state=SEED)"
609
+ ]
610
+ },
611
+ {
612
+ "cell_type": "markdown",
613
+ "metadata": {},
614
+ "source": [
615
+ "### Processando os dados\n",
616
+ "A função process_data recebe uma linha de um dataframe contendo uma revisΓ£o de texto e sua respectiva classificação de sentimento. Ela comeΓ§a extraindo e limpando o texto da revisΓ£o, removendo quaisquer espaΓ§os extras. Em seguida, utiliza o tokenizer BERT para tokenizar o texto, aplicando padding e truncamento para garantir que todas as sequΓͺncias tenham um comprimento fixo definido pela variΓ‘vel MAX_LEN. A função entΓ£o adiciona a etiqueta de sentimento original e o texto limpo Γ s codificaçáes geradas, retornando um dicionΓ‘rio que contΓ©m os tokens do texto, a etiqueta de sentimento e o texto original."
617
+ ]
618
+ },
619
+ {
620
+ "cell_type": "code",
621
+ "execution_count": 16,
622
+ "metadata": {
623
+ "id": "v7EZ6wd-qDfd"
624
+ },
625
+ "outputs": [],
626
+ "source": [
627
+ "def process_data(row):\n",
628
+ "\n",
629
+ " text = row['review']\n",
630
+ " text = str(text)\n",
631
+ " text = ' '.join(text.split())\n",
632
+ "\n",
633
+ " encodings = tokenizer(text, padding=\"max_length\", truncation=True, max_length=MAX_LEN)\n",
634
+ "\n",
635
+ " encodings['label'] = row['sentiment']\n",
636
+ " encodings['text'] = text\n",
637
+ "\n",
638
+ " return encodings"
639
+ ]
640
+ },
641
+ {
642
+ "cell_type": "code",
643
+ "execution_count": 17,
644
+ "metadata": {
645
+ "id": "d9VgrXNSqIYL"
646
+ },
647
+ "outputs": [],
648
+ "source": [
649
+ "# Treino\n",
650
+ "processed_data_tr = []\n",
651
+ "for i in range(df_train.shape[0]):\n",
652
+ " processed_data_tr.append(process_data(df_train.iloc[i]))"
653
+ ]
654
+ },
655
+ {
656
+ "cell_type": "code",
657
+ "execution_count": 18,
658
+ "metadata": {
659
+ "id": "p0NLQxoKqJ_k"
660
+ },
661
+ "outputs": [],
662
+ "source": [
663
+ "# Validação\n",
664
+ "processed_data_val = []\n",
665
+ "for i in range(df_val.shape[0]):\n",
666
+ " processed_data_val.append(process_data(df_val.iloc[i]))"
667
+ ]
668
+ },
669
+ {
670
+ "cell_type": "code",
671
+ "execution_count": 19,
672
+ "metadata": {
673
+ "id": "ac76Rb6fqP_G"
674
+ },
675
+ "outputs": [],
676
+ "source": [
677
+ "# Dataframes de Treino e Validação\n",
678
+ "df_train = pd.DataFrame(processed_data_tr)\n",
679
+ "df_val = pd.DataFrame(processed_data_val)"
680
+ ]
681
+ },
682
+ {
683
+ "cell_type": "code",
684
+ "execution_count": 20,
685
+ "metadata": {
686
+ "colab": {
687
+ "base_uri": "https://localhost:8080/",
688
+ "height": 206
689
+ },
690
+ "id": "RdbHaVy_fd64",
691
+ "outputId": "a9aed834-81b7-4223-da42-6289799c2e1e"
692
+ },
693
+ "outputs": [
694
+ {
695
+ "data": {
696
+ "text/html": [
697
+ "<div>\n",
698
+ "<style scoped>\n",
699
+ " .dataframe tbody tr th:only-of-type {\n",
700
+ " vertical-align: middle;\n",
701
+ " }\n",
702
+ "\n",
703
+ " .dataframe tbody tr th {\n",
704
+ " vertical-align: top;\n",
705
+ " }\n",
706
+ "\n",
707
+ " .dataframe thead th {\n",
708
+ " text-align: right;\n",
709
+ " }\n",
710
+ "</style>\n",
711
+ "<table border=\"1\" class=\"dataframe\">\n",
712
+ " <thead>\n",
713
+ " <tr style=\"text-align: right;\">\n",
714
+ " <th></th>\n",
715
+ " <th>attention_mask</th>\n",
716
+ " <th>input_ids</th>\n",
717
+ " <th>label</th>\n",
718
+ " <th>text</th>\n",
719
+ " <th>token_type_ids</th>\n",
720
+ " </tr>\n",
721
+ " </thead>\n",
722
+ " <tbody>\n",
723
+ " <tr>\n",
724
+ " <th>0</th>\n",
725
+ " <td>[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ...</td>\n",
726
+ " <td>[101, 2921, 3198, 23624, 2954, 6978, 2674, 841...</td>\n",
727
+ " <td>0</td>\n",
728
+ " <td>kept ask mani fight scream match swear gener m...</td>\n",
729
+ " <td>[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...</td>\n",
730
+ " </tr>\n",
731
+ " <tr>\n",
732
+ " <th>1</th>\n",
733
+ " <td>[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ...</td>\n",
734
+ " <td>[101, 3422, 4372, 3775, 2099, 9587, 5737, 2071...</td>\n",
735
+ " <td>0</td>\n",
736
+ " <td>watch entir movi could watch entir movi stop d...</td>\n",
737
+ " <td>[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...</td>\n",
738
+ " </tr>\n",
739
+ " <tr>\n",
740
+ " <th>2</th>\n",
741
+ " <td>[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ...</td>\n",
742
+ " <td>[101, 3543, 2293, 2358, 10050, 2128, 25300, 11...</td>\n",
743
+ " <td>1</td>\n",
744
+ " <td>touch love stori reminisc Β‘in mood love draw h...</td>\n",
745
+ " <td>[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...</td>\n",
746
+ " </tr>\n",
747
+ " <tr>\n",
748
+ " <th>3</th>\n",
749
+ " <td>[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ...</td>\n",
750
+ " <td>[101, 3732, 2154, 11865, 15472, 2072, 8040, 73...</td>\n",
751
+ " <td>0</td>\n",
752
+ " <td>latter day fulci schlocker total abysm concoct...</td>\n",
753
+ " <td>[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...</td>\n",
754
+ " </tr>\n",
755
+ " <tr>\n",
756
+ " <th>4</th>\n",
757
+ " <td>[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ...</td>\n",
758
+ " <td>[101, 2034, 3813, 3669, 19337, 2666, 2615, 504...</td>\n",
759
+ " <td>0</td>\n",
760
+ " <td>first firmli believ norwegian movi continu get...</td>\n",
761
+ " <td>[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...</td>\n",
762
+ " </tr>\n",
763
+ " </tbody>\n",
764
+ "</table>\n",
765
+ "</div>"
766
+ ],
767
+ "text/plain": [
768
+ " attention_mask \\\n",
769
+ "0 [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ... \n",
770
+ "1 [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ... \n",
771
+ "2 [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ... \n",
772
+ "3 [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ... \n",
773
+ "4 [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ... \n",
774
+ "\n",
775
+ " input_ids label \\\n",
776
+ "0 [101, 2921, 3198, 23624, 2954, 6978, 2674, 841... 0 \n",
777
+ "1 [101, 3422, 4372, 3775, 2099, 9587, 5737, 2071... 0 \n",
778
+ "2 [101, 3543, 2293, 2358, 10050, 2128, 25300, 11... 1 \n",
779
+ "3 [101, 3732, 2154, 11865, 15472, 2072, 8040, 73... 0 \n",
780
+ "4 [101, 2034, 3813, 3669, 19337, 2666, 2615, 504... 0 \n",
781
+ "\n",
782
+ " text \\\n",
783
+ "0 kept ask mani fight scream match swear gener m... \n",
784
+ "1 watch entir movi could watch entir movi stop d... \n",
785
+ "2 touch love stori reminisc Β‘in mood love draw h... \n",
786
+ "3 latter day fulci schlocker total abysm concoct... \n",
787
+ "4 first firmli believ norwegian movi continu get... \n",
788
+ "\n",
789
+ " token_type_ids \n",
790
+ "0 [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ... \n",
791
+ "1 [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ... \n",
792
+ "2 [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ... \n",
793
+ "3 [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ... \n",
794
+ "4 [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ... "
795
+ ]
796
+ },
797
+ "execution_count": 20,
798
+ "metadata": {},
799
+ "output_type": "execute_result"
800
+ }
801
+ ],
802
+ "source": [
803
+ "df_train.head()"
804
+ ]
805
+ },
806
+ {
807
+ "cell_type": "markdown",
808
+ "metadata": {
809
+ "id": "0lTWT8JwkRic"
810
+ },
811
+ "source": [
812
+ "## Fine Tunning do Modelo\n",
813
+ "Ajuste fino do BERT para tarefas específica de classificação de sentimento para o dataset do IMDB"
814
+ ]
815
+ },
816
+ {
817
+ "cell_type": "code",
818
+ "execution_count": null,
819
+ "metadata": {},
820
+ "outputs": [],
821
+ "source": [
822
+ "import torch\n",
823
+ "import pyarrow as pa\n",
824
+ "from datasets import Dataset\n",
825
+ "import evaluate\n",
826
+ "import numpy as np"
827
+ ]
828
+ },
829
+ {
830
+ "cell_type": "code",
831
+ "execution_count": 21,
832
+ "metadata": {
833
+ "colab": {
834
+ "base_uri": "https://localhost:8080/"
835
+ },
836
+ "id": "kW53p7VQqUDD",
837
+ "outputId": "8231f3ba-37d5-4546-c4d0-6b4ff317ecf3"
838
+ },
839
+ "outputs": [
840
+ {
841
+ "data": {
842
+ "text/plain": [
843
+ "device(type='cuda', index=0)"
844
+ ]
845
+ },
846
+ "execution_count": 21,
847
+ "metadata": {},
848
+ "output_type": "execute_result"
849
+ }
850
+ ],
851
+ "source": [
852
+ "device = torch.device(\"cuda:0\" if torch.cuda.is_available() else \"cpu\")\n",
853
+ "device"
854
+ ]
855
+ },
856
+ {
857
+ "cell_type": "code",
858
+ "execution_count": 24,
859
+ "metadata": {
860
+ "id": "68OdbTv5rLrm"
861
+ },
862
+ "outputs": [],
863
+ "source": [
864
+ "train_hg = Dataset(pa.Table.from_pandas(df_train))\n",
865
+ "valid_hg = Dataset(pa.Table.from_pandas(df_val))"
866
+ ]
867
+ },
868
+ {
869
+ "cell_type": "markdown",
870
+ "metadata": {},
871
+ "source": [
872
+ "## Metricas de avaliação F1 Score e Acc"
873
+ ]
874
+ },
875
+ {
876
+ "cell_type": "markdown",
877
+ "metadata": {},
878
+ "source": [
879
+ "`compute_metrics` calcula tanto a acurÑcia quanto o F1-score para avaliar um modelo de classificação. Primeiramente, são carregadas as métricas de acurÑcia e F1-score usando evaluate.load. Em seguida, a função compute_metrics recebe um par de arrays eval_pred, contendo as previsáes do modelo e os rótulos verdadeiros. Utilizando as previsáes, a função calcula a acurÑcia e o F1-score ponderado, onde a acurÑcia é obtida através da comparação das previsáes com os rótulos utilizando a métrica de acurÑcia previamente carregada, e o F1-score é calculado utilizando a métrica de F1 previamente carregada, com ponderação \"weighted\". Os resultados de ambas as métricas são então combinados em um dicionÑrio e retornados como um único objeto contendo as métricas de avaliação calculadas."
880
+ ]
881
+ },
882
+ {
883
+ "cell_type": "code",
884
+ "execution_count": 25,
885
+ "metadata": {
886
+ "id": "lUNhDPs0ry4m"
887
+ },
888
+ "outputs": [],
889
+ "source": [
890
+ "\n",
891
+ "# Load both accuracy and f1 metrics\n",
892
+ "accuracy_metric = evaluate.load(\"accuracy\")\n",
893
+ "f1_metric = evaluate.load(\"f1\")\n",
894
+ "\n",
895
+ "# Metric helper method\n",
896
+ "def compute_metrics(eval_pred):\n",
897
+ " predictions, labels = eval_pred\n",
898
+ " predictions = np.argmax(predictions, axis=1)\n",
899
+ "\n",
900
+ " # Compute accuracy\n",
901
+ " accuracy = accuracy_metric.compute(predictions=predictions, references=labels)\n",
902
+ "\n",
903
+ " # Compute F1 score\n",
904
+ " f1 = f1_metric.compute(predictions=predictions, references=labels, average=\"weighted\")\n",
905
+ "\n",
906
+ " # Combine the metrics into a single dictionary\n",
907
+ " combined_metrics = {\n",
908
+ " 'accuracy': accuracy['accuracy'],\n",
909
+ " 'f1': f1['f1']\n",
910
+ " }\n",
911
+ "\n",
912
+ " return combined_metrics"
913
+ ]
914
+ },
915
+ {
916
+ "cell_type": "code",
917
+ "execution_count": 26,
918
+ "metadata": {
919
+ "colab": {
920
+ "base_uri": "https://localhost:8080/"
921
+ },
922
+ "id": "9jJYTWsHjnEc",
923
+ "outputId": "fe45691a-4476-4978-89b8-15f36465c37c"
924
+ },
925
+ "outputs": [
926
+ {
927
+ "name": "stdout",
928
+ "output_type": "stream",
929
+ "text": [
930
+ "Name: accelerateNote: you may need to restart the kernel to use updated packages.\n",
931
+ "\n",
932
+ "Version: 0.31.0\n",
933
+ "Summary: Accelerate\n",
934
+ "Home-page: https://github.com/huggingface/accelerate\n",
935
+ "Author: The HuggingFace team\n",
936
+ "Author-email: [email protected]\n",
937
+ "License: Apache\n",
938
+ "Location: c:\\Users\\andre\\1JUPYTER\\dt_labs\\.venv\\Lib\\site-packages\n",
939
+ "Requires: huggingface-hub, numpy, packaging, psutil, pyyaml, safetensors, torch\n",
940
+ "Required-by: \n",
941
+ "---\n",
942
+ "Name: transformers\n",
943
+ "Version: 4.41.2\n",
944
+ "Summary: State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow\n",
945
+ "Home-page: https://github.com/huggingface/transformers\n",
946
+ "Author: The Hugging Face team (past and future) with the help of all our contributors (https://github.com/huggingface/transformers/graphs/contributors)\n",
947
+ "Author-email: [email protected]\n",
948
+ "License: Apache 2.0 License\n",
949
+ "Location: c:\\Users\\andre\\1JUPYTER\\dt_labs\\.venv\\Lib\\site-packages\n",
950
+ "Requires: filelock, huggingface-hub, numpy, packaging, pyyaml, regex, requests, safetensors, tokenizers, tqdm\n",
951
+ "Required-by: \n"
952
+ ]
953
+ }
954
+ ],
955
+ "source": [
956
+ "pip show accelerate transformers"
957
+ ]
958
+ },
959
+ {
960
+ "cell_type": "markdown",
961
+ "metadata": {},
962
+ "source": [
963
+ "## Treinamento do modelo"
964
+ ]
965
+ },
966
+ {
967
+ "cell_type": "code",
968
+ "execution_count": 27,
969
+ "metadata": {
970
+ "colab": {
971
+ "base_uri": "https://localhost:8080/"
972
+ },
973
+ "id": "QlaLCwf7rLtp",
974
+ "outputId": "7e10e82a-8bc7-478b-851e-c7b628b46c41"
975
+ },
976
+ "outputs": [
977
+ {
978
+ "name": "stderr",
979
+ "output_type": "stream",
980
+ "text": [
981
+ "c:\\Users\\andre\\1JUPYTER\\dt_labs\\.venv\\Lib\\site-packages\\transformers\\training_args.py:1474: FutureWarning: `evaluation_strategy` is deprecated and will be removed in version 4.46 of πŸ€— Transformers. Use `eval_strategy` instead\n",
982
+ " warnings.warn(\n"
983
+ ]
984
+ }
985
+ ],
986
+ "source": [
987
+ "from transformers import TrainingArguments, Trainer\n",
988
+ "\n",
989
+ "EPOCHS = 1\n",
990
+ "\n",
991
+ "training_args = TrainingArguments(output_dir=\"./result\",\n",
992
+ " evaluation_strategy=\"epoch\",\n",
993
+ " num_train_epochs= EPOCHS,\n",
994
+ " per_device_train_batch_size=16,\n",
995
+ " per_device_eval_batch_size=8\n",
996
+ " )\n",
997
+ "\n",
998
+ "trainer = Trainer(\n",
999
+ " model=model,\n",
1000
+ " args=training_args,\n",
1001
+ " train_dataset=train_hg,\n",
1002
+ " eval_dataset=valid_hg,\n",
1003
+ " tokenizer=tokenizer,\n",
1004
+ " compute_metrics=compute_metrics\n",
1005
+ ")"
1006
+ ]
1007
+ },
1008
+ {
1009
+ "cell_type": "code",
1010
+ "execution_count": 28,
1011
+ "metadata": {},
1012
+ "outputs": [
1013
+ {
1014
+ "name": "stdout",
1015
+ "output_type": "stream",
1016
+ "text": [
1017
+ "CUDA available: True\n",
1018
+ "CUDA version: 12.1\n"
1019
+ ]
1020
+ }
1021
+ ],
1022
+ "source": [
1023
+ "print(\"CUDA available: \", torch.cuda.is_available())\n",
1024
+ "print(\"CUDA version: \", torch.version.cuda)"
1025
+ ]
1026
+ },
1027
+ {
1028
+ "cell_type": "code",
1029
+ "execution_count": 29,
1030
+ "metadata": {
1031
+ "colab": {
1032
+ "base_uri": "https://localhost:8080/",
1033
+ "height": 141
1034
+ },
1035
+ "id": "3s6lVFz_rLwO",
1036
+ "outputId": "ee64e8e9-9c8c-42a8-c355-f51410cc33df"
1037
+ },
1038
+ "outputs": [
1039
+ {
1040
+ "name": "stderr",
1041
+ "output_type": "stream",
1042
+ "text": [
1043
+ " 0%| | 0/2500 [00:00<?, ?it/s]c:\\Users\\andre\\1JUPYTER\\dt_labs\\.venv\\Lib\\site-packages\\transformers\\models\\bert\\modeling_bert.py:435: UserWarning: 1Torch was not compiled with flash attention. (Triggered internally at ..\\aten\\src\\ATen\\native\\transformers\\cuda\\sdp_utils.cpp:263.)\n",
1044
+ " attn_output = torch.nn.functional.scaled_dot_product_attention(\n",
1045
+ " 20%|β–ˆβ–ˆ | 500/2500 [05:35<22:22, 1.49it/s]"
1046
+ ]
1047
+ },
1048
+ {
1049
+ "name": "stdout",
1050
+ "output_type": "stream",
1051
+ "text": [
1052
+ "{'loss': 0.4994, 'grad_norm': 12.613661766052246, 'learning_rate': 4e-05, 'epoch': 0.2}\n"
1053
+ ]
1054
+ },
1055
+ {
1056
+ "name": "stderr",
1057
+ "output_type": "stream",
1058
+ "text": [
1059
+ " 40%|β–ˆβ–ˆβ–ˆβ–ˆ | 1000/2500 [11:13<16:46, 1.49it/s]"
1060
+ ]
1061
+ },
1062
+ {
1063
+ "name": "stdout",
1064
+ "output_type": "stream",
1065
+ "text": [
1066
+ "{'loss': 0.3898, 'grad_norm': 4.661791801452637, 'learning_rate': 3e-05, 'epoch': 0.4}\n"
1067
+ ]
1068
+ },
1069
+ {
1070
+ "name": "stderr",
1071
+ "output_type": "stream",
1072
+ "text": [
1073
+ " 60%|β–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆ | 1500/2500 [16:47<11:02, 1.51it/s]"
1074
+ ]
1075
+ },
1076
+ {
1077
+ "name": "stdout",
1078
+ "output_type": "stream",
1079
+ "text": [
1080
+ "{'loss': 0.3516, 'grad_norm': 1.5203113555908203, 'learning_rate': 2e-05, 'epoch': 0.6}\n"
1081
+ ]
1082
+ },
1083
+ {
1084
+ "name": "stderr",
1085
+ "output_type": "stream",
1086
+ "text": [
1087
+ " 80%|β–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆ | 2000/2500 [22:25<05:33, 1.50it/s]"
1088
+ ]
1089
+ },
1090
+ {
1091
+ "name": "stdout",
1092
+ "output_type": "stream",
1093
+ "text": [
1094
+ "{'loss': 0.3121, 'grad_norm': 8.331348419189453, 'learning_rate': 1e-05, 'epoch': 0.8}\n"
1095
+ ]
1096
+ },
1097
+ {
1098
+ "name": "stderr",
1099
+ "output_type": "stream",
1100
+ "text": [
1101
+ "100%|β–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆ| 2500/2500 [28:04<00:00, 1.50it/s]"
1102
+ ]
1103
+ },
1104
+ {
1105
+ "name": "stdout",
1106
+ "output_type": "stream",
1107
+ "text": [
1108
+ "{'loss': 0.2882, 'grad_norm': 6.287994861602783, 'learning_rate': 0.0, 'epoch': 1.0}\n"
1109
+ ]
1110
+ },
1111
+ {
1112
+ "name": "stderr",
1113
+ "output_type": "stream",
1114
+ "text": [
1115
+ " \n",
1116
+ "100%|β–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆ| 2500/2500 [30:45<00:00, 1.35it/s]"
1117
+ ]
1118
+ },
1119
+ {
1120
+ "name": "stdout",
1121
+ "output_type": "stream",
1122
+ "text": [
1123
+ "{'eval_loss': 0.283893883228302, 'eval_accuracy': 0.883, 'eval_f1': 0.8829425082505502, 'eval_runtime': 159.717, 'eval_samples_per_second': 62.611, 'eval_steps_per_second': 7.826, 'epoch': 1.0}\n",
1124
+ "{'train_runtime': 1845.2907, 'train_samples_per_second': 21.677, 'train_steps_per_second': 1.355, 'train_loss': 0.3682089477539062, 'epoch': 1.0}\n"
1125
+ ]
1126
+ },
1127
+ {
1128
+ "name": "stderr",
1129
+ "output_type": "stream",
1130
+ "text": [
1131
+ "\n"
1132
+ ]
1133
+ },
1134
+ {
1135
+ "data": {
1136
+ "text/plain": [
1137
+ "TrainOutput(global_step=2500, training_loss=0.3682089477539062, metrics={'train_runtime': 1845.2907, 'train_samples_per_second': 21.677, 'train_steps_per_second': 1.355, 'total_flos': 4111110240000000.0, 'train_loss': 0.3682089477539062, 'epoch': 1.0})"
1138
+ ]
1139
+ },
1140
+ "execution_count": 29,
1141
+ "metadata": {},
1142
+ "output_type": "execute_result"
1143
+ }
1144
+ ],
1145
+ "source": [
1146
+ "trainer.train()"
1147
+ ]
1148
+ },
1149
+ {
1150
+ "cell_type": "markdown",
1151
+ "metadata": {},
1152
+ "source": [
1153
+ "## Salvando o modelo"
1154
+ ]
1155
+ },
1156
+ {
1157
+ "cell_type": "code",
1158
+ "execution_count": 38,
1159
+ "metadata": {
1160
+ "id": "8eO6WDiOBAhg"
1161
+ },
1162
+ "outputs": [],
1163
+ "source": [
1164
+ "torch.save(model.state_dict(), 'model.pth')"
1165
+ ]
1166
+ },
1167
+ {
1168
+ "cell_type": "markdown",
1169
+ "metadata": {
1170
+ "id": "FtVZztSa40b3"
1171
+ },
1172
+ "source": [
1173
+ "## Teste de prediçáes individuais"
1174
+ ]
1175
+ },
1176
+ {
1177
+ "cell_type": "code",
1178
+ "execution_count": 34,
1179
+ "metadata": {
1180
+ "id": "lOHVSyfJJ8zK"
1181
+ },
1182
+ "outputs": [],
1183
+ "source": [
1184
+ "from transformers import AutoTokenizer\n",
1185
+ "\n",
1186
+ "new_tokenizer = AutoTokenizer.from_pretrained(pre_trained_base)"
1187
+ ]
1188
+ },
1189
+ {
1190
+ "cell_type": "code",
1191
+ "execution_count": 35,
1192
+ "metadata": {
1193
+ "id": "t-T7hDZ2J1Qk"
1194
+ },
1195
+ "outputs": [],
1196
+ "source": [
1197
+ "def get_prediction(text):\n",
1198
+ " encoding = new_tokenizer(text, return_tensors=\"pt\", padding=\"max_length\", truncation=True, max_length=MAX_LEN)\n",
1199
+ " encoding = {k: v.to(trainer.model.device) for k,v in encoding.items()}\n",
1200
+ "\n",
1201
+ " outputs = model(**encoding)\n",
1202
+ "\n",
1203
+ " logits = outputs.logits\n",
1204
+ "\n",
1205
+ " sigmoid = torch.nn.Sigmoid()\n",
1206
+ " probs = sigmoid(logits.squeeze().cpu())\n",
1207
+ " probs = probs.detach().numpy()\n",
1208
+ " label = np.argmax(probs, axis=-1)\n",
1209
+ "\n",
1210
+ " return label"
1211
+ ]
1212
+ },
1213
+ {
1214
+ "cell_type": "code",
1215
+ "execution_count": 36,
1216
+ "metadata": {
1217
+ "colab": {
1218
+ "base_uri": "https://localhost:8080/"
1219
+ },
1220
+ "id": "y4dxQ4oYJ5C1",
1221
+ "outputId": "d0d77c2d-aff6-412b-e22a-0b721f5b097e"
1222
+ },
1223
+ "outputs": [
1224
+ {
1225
+ "data": {
1226
+ "text/plain": [
1227
+ "0"
1228
+ ]
1229
+ },
1230
+ "execution_count": 36,
1231
+ "metadata": {},
1232
+ "output_type": "execute_result"
1233
+ }
1234
+ ],
1235
+ "source": [
1236
+ "get_prediction(\"This movie is horrible!\")"
1237
+ ]
1238
+ },
1239
+ {
1240
+ "cell_type": "code",
1241
+ "execution_count": 37,
1242
+ "metadata": {
1243
+ "colab": {
1244
+ "base_uri": "https://localhost:8080/"
1245
+ },
1246
+ "id": "JXAyOu_6AqoO",
1247
+ "outputId": "ffcd019e-4c0c-45eb-f538-d2860c53a0e0"
1248
+ },
1249
+ "outputs": [
1250
+ {
1251
+ "data": {
1252
+ "text/plain": [
1253
+ "1"
1254
+ ]
1255
+ },
1256
+ "execution_count": 37,
1257
+ "metadata": {},
1258
+ "output_type": "execute_result"
1259
+ }
1260
+ ],
1261
+ "source": [
1262
+ "get_prediction(\"This movie is awesome!\")"
1263
+ ]
1264
+ }
1265
+ ],
1266
+ "metadata": {
1267
+ "accelerator": "GPU",
1268
+ "colab": {
1269
+ "provenance": []
1270
+ },
1271
+ "gpuClass": "standard",
1272
+ "kernelspec": {
1273
+ "display_name": "Python 3",
1274
+ "name": "python3"
1275
+ },
1276
+ "language_info": {
1277
+ "codemirror_mode": {
1278
+ "name": "ipython",
1279
+ "version": 3
1280
+ },
1281
+ "file_extension": ".py",
1282
+ "mimetype": "text/x-python",
1283
+ "name": "python",
1284
+ "nbconvert_exporter": "python",
1285
+ "pygments_lexer": "ipython3",
1286
+ "version": "3.10.11"
1287
+ }
1288
+ },
1289
+ "nbformat": 4,
1290
+ "nbformat_minor": 0
1291
+ }
notebooks_explicativos/Simbolico.ipynb ADDED
The diff for this file is too large to render. See raw diff