Merge pull request #8715 from berak:fix_digits_py

This commit is contained in:
Maksim Shabunin 2017-05-11 12:46:41 +00:00
commit aaabb07bee

View File

@ -109,7 +109,7 @@ def evaluate_model(model, digits, samples, labels):
confusion = np.zeros((10, 10), np.int32) confusion = np.zeros((10, 10), np.int32)
for i, j in zip(labels, resp): for i, j in zip(labels, resp):
confusion[i, j] += 1 confusion[i, int(j)] += 1
print('confusion matrix:') print('confusion matrix:')
print(confusion) print(confusion)
print() print()