Skip to content

Commit f75c48b

Browse files
Add test_dir variable for path to testing directory
1 parent 893c764 commit f75c48b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

site/en/tutorials/keras/text_classification.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,6 +194,7 @@
194194
"outputs": [],
195195
"source": [
196196
"train_dir = os.path.join(dataset_dir, 'aclImdb', 'train')\n",
197+
"test_dir = os.path.join(dataset_dir, 'aclImdb', 'test')\n",
197198
"os.listdir(train_dir)"
198199
]
199200
},
@@ -382,7 +383,7 @@
382383
"outputs": [],
383384
"source": [
384385
"raw_test_ds = tf.keras.utils.text_dataset_from_directory(\n",
385-
" 'aclImdb/test',\n",
386+
" test_dir,\n",
386387
" batch_size=batch_size)"
387388
]
388389
},

0 commit comments

Comments
 (0)