Skip to content

Commit 50a83a2

Browse files
authored
Merge pull request jdwittenauer#5 from yossiz74/patch-1
open pickle files in 'rb' mode
2 parents f93e931 + 6b2f1c7 commit 50a83a2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

notebooks/tensorflow/Tensorflow-1-NotMNIST.ipynb

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -449,7 +449,7 @@
449449
}
450450
],
451451
"source": [
452-
"file = open(train_datasets[0])\n",
452+
"file = open(train_datasets[0], 'rb')\n",
453453
"train_A = pickle.load(file)\n",
454454
"train_A.shape"
455455
]
@@ -538,11 +538,11 @@
538538
"source": [
539539
"print('Checking number of images per class...')\n",
540540
"for letter in train_datasets:\n",
541-
" file = open(letter)\n",
541+
" file = open(letter, 'rb')\n",
542542
" dataset = pickle.load(file)\n",
543543
" print(letter + ' size: ' + str(dataset.shape[0]))\n",
544544
"for letter in test_datasets:\n",
545-
" file = open(letter)\n",
545+
" file = open(letter, 'rb')\n",
546546
" dataset = pickle.load(file)\n",
547547
" print(letter + ' size: ' + str(dataset.shape[0]))"
548548
]

0 commit comments

Comments
 (0)