Commit 7b05d52f authored by berak's avatar berak

py_samples: fix digits.py

parent 06b0fe35
......@@ -109,7 +109,7 @@ def evaluate_model(model, digits, samples, labels):
confusion = np.zeros((10, 10), np.int32)
for i, j in zip(labels, resp):
confusion[i, j] += 1
confusion[i, int(j)] += 1
print('confusion matrix:')
print(confusion)
print()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment