@@ -33,8 +33,8 @@ def main():
33
33
# correctly
34
34
num_correct = sum (int (guess_digit (image , avgs ) == digit )
35
35
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 ]) ))
38
38
39
39
def avg_darknesses (training_data ):
40
40
""" Return a defaultdict whose keys are the digits 0 through 9.
@@ -47,7 +47,7 @@ def avg_darknesses(training_data):
47
47
digit_counts [digit ] += 1
48
48
darknesses [digit ] += sum (image )
49
49
avgs = defaultdict (float )
50
- for digit , n in digit_counts .iteritems ():
50
+ for digit , n in digit_counts .items ():
51
51
avgs [digit ] = darknesses [digit ] / n
52
52
return avgs
53
53
@@ -57,7 +57,7 @@ def guess_digit(image, avgs):
57
57
assumed to be a defaultdict whose keys are 0...9, and whose values
58
58
are the corresponding average darknesses across the training data."""
59
59
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 ()}
61
61
return min (distances , key = distances .get )
62
62
63
63
if __name__ == "__main__" :
0 commit comments