Skip to content

Commit 2228ee4

Browse files
authored
Merge pull request #7 from yossiz74/patch-3
Don't modify test_dataset when training
2 parents ea26562 + b224be6 commit 2228ee4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

notebooks/tensorflow/Tensorflow-1-NotMNIST.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -935,13 +935,13 @@
935935
"train_sample = np.reshape(train_sample, (samples, width * height))\n",
936936
"\n",
937937
"(samples, width, height) = test_dataset.shape\n",
938-
"test_dataset = np.reshape(test_dataset, (samples, width * height))\n",
938+
"test_dataset_reshaped = np.reshape(test_dataset, (samples, width * height))\n",
939939
"\n",
940940
"model = LogisticRegression(penalty='l2', C=1.0)\n",
941941
"model.fit(train_sample, train_sample_labels)\n",
942942
"\n",
943943
"train_score = model.score(train_sample, train_sample_labels)\n",
944-
"test_score = model.score(test_dataset, test_labels)\n",
944+
"test_score = model.score(test_dataset_reshaped, test_labels)\n",
945945
"print('Training score = ', train_score)\n",
946946
"print('Test score = ', test_score)"
947947
]

0 commit comments

Comments
 (0)