|
34 | 34 | "import nltk\n", |
35 | 35 | "nltk.download('stopwords')" |
36 | 36 | ], |
37 | | - "execution_count": 1, |
| 37 | + "execution_count": 0, |
38 | 38 | "outputs": [ |
39 | 39 | { |
40 | 40 | "output_type": "stream", |
|
107 | 107 | "device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')\n", |
108 | 108 | "device" |
109 | 109 | ], |
110 | | - "execution_count": 2, |
| 110 | + "execution_count": 0, |
111 | 111 | "outputs": [ |
112 | 112 | { |
113 | 113 | "output_type": "execute_result", |
|
133 | 133 | "# RNN Text Classification: Predict the sentiment of IMDB movie reviews" |
134 | 134 | ] |
135 | 135 | }, |
136 | | - { |
137 | | - "metadata": { |
138 | | - "colab_type": "text", |
139 | | - "id": "lcDm8cJ6KIgt" |
140 | | - }, |
141 | | - "cell_type": "markdown", |
142 | | - "source": [ |
143 | | - "## Specify Hyperparameters" |
144 | | - ] |
145 | | - }, |
146 | | - { |
147 | | - "metadata": { |
148 | | - "colab_type": "code", |
149 | | - "id": "F4z4mY2kKGlq", |
150 | | - "scrolled": true, |
151 | | - "colab": {} |
152 | | - }, |
153 | | - "cell_type": "code", |
154 | | - "source": [ |
155 | | - "DROPOUT_PROBABILITY = 0.3\n", |
156 | | - "LEARNING_RATE = 1e-3\n", |
157 | | - "N_RNN_LAYERS = 1\n", |
158 | | - "EMBEDDING_DIMENSION = 128\n", |
159 | | - "HIDDEN_SIZE = 64\n", |
160 | | - "IS_BIDIRECTIONAL = True" |
161 | | - ], |
162 | | - "execution_count": 0, |
163 | | - "outputs": [] |
164 | | - }, |
165 | 136 | { |
166 | 137 | "metadata": { |
167 | 138 | "colab_type": "text", |
|
296 | 267 | "\n", |
297 | 268 | "dataset = ImdbSentimentDataset(DATA_PATH, max_vocab, max_len)" |
298 | 269 | ], |
299 | | - "execution_count": 5, |
| 270 | + "execution_count": 0, |
300 | 271 | "outputs": [ |
301 | 272 | { |
302 | 273 | "output_type": "stream", |
|
362 | 333 | " dataset, valid_ratio=0.05, test_ratio=0.05)\n", |
363 | 334 | "len(train_dataset), len(valid_dataset), len(test_dataset)" |
364 | 335 | ], |
365 | | - "execution_count": 7, |
| 336 | + "execution_count": 0, |
366 | 337 | "outputs": [ |
367 | 338 | { |
368 | 339 | "output_type": "execute_result", |
|
592 | 563 | "model = model.to(device)\n", |
593 | 564 | "model" |
594 | 565 | ], |
595 | | - "execution_count": 12, |
| 566 | + "execution_count": 0, |
596 | 567 | "outputs": [ |
597 | 568 | { |
598 | 569 | "output_type": "execute_result", |
|
748 | 719 | " \n", |
749 | 720 | " n_epochs += 1" |
750 | 721 | ], |
751 | | - "execution_count": 15, |
| 722 | + "execution_count": 0, |
752 | 723 | "outputs": [ |
753 | 724 | { |
754 | 725 | "output_type": "display_data", |
|
1339 | 1310 | "plt.xticks(epoch_ticks)\n", |
1340 | 1311 | "plt.show()" |
1341 | 1312 | ], |
1342 | | - "execution_count": 16, |
| 1313 | + "execution_count": 0, |
1343 | 1314 | "outputs": [ |
1344 | 1315 | { |
1345 | 1316 | "output_type": "display_data", |
|
1409 | 1380 | " \n", |
1410 | 1381 | "print(classification_report(y_true, y_pred))" |
1411 | 1382 | ], |
1412 | | - "execution_count": 22, |
| 1383 | + "execution_count": 0, |
1413 | 1384 | "outputs": [ |
1414 | 1385 | { |
1415 | 1386 | "output_type": "display_data", |
|
1514 | 1485 | "source": [ |
1515 | 1486 | "print_random_prediction(n=5)" |
1516 | 1487 | ], |
1517 | | - "execution_count": 23, |
| 1488 | + "execution_count": 0, |
1518 | 1489 | "outputs": [ |
1519 | 1490 | { |
1520 | 1491 | "output_type": "display_data", |
|
0 commit comments