Skip to content

Commit f48defe

Browse files
Merge pull request MichalDanielDobrzanski#1 from anandman/master
updated mnist_average_darkness.py for Python 3.5.2 - further improvements
2 parents e99416e + 3ef0f0b commit f48defe

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

mnist_average_darkness.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,8 @@ def main():
3333
# correctly
3434
num_correct = sum(int(guess_digit(image, avgs) == digit)
3535
for image, digit in zip(test_data[0], test_data[1]))
36-
print "Baseline classifier using average darkness of image."
37-
print "%s of %s values correct." % (num_correct, len(test_data[1]))
36+
print("Baseline classifier using average darkness of image.")
37+
print("{0} of {1} values correct.".format(num_correct, len(test_data[1])))
3838

3939
def avg_darknesses(training_data):
4040
""" Return a defaultdict whose keys are the digits 0 through 9.
@@ -47,7 +47,7 @@ def avg_darknesses(training_data):
4747
digit_counts[digit] += 1
4848
darknesses[digit] += sum(image)
4949
avgs = defaultdict(float)
50-
for digit, n in digit_counts.iteritems():
50+
for digit, n in digit_counts.items():
5151
avgs[digit] = darknesses[digit] / n
5252
return avgs
5353

@@ -57,7 +57,7 @@ def guess_digit(image, avgs):
5757
assumed to be a defaultdict whose keys are 0...9, and whose values
5858
are the corresponding average darknesses across the training data."""
5959
darkness = sum(image)
60-
distances = {k: abs(v-darkness) for k, v in avgs.iteritems()}
60+
distances = {k: abs(v-darkness) for k, v in avgs.items()}
6161
return min(distances, key=distances.get)
6262

6363
if __name__ == "__main__":

0 commit comments

Comments
 (0)