Skip to content

Commit

Permalink
Always print non-normalised matrix
Browse files Browse the repository at this point in the history
  • Loading branch information
msoos committed May 23, 2018
1 parent 40f5e0a commit 7307016
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions scripts/learn/mini_pred.py
Original file line number Diff line number Diff line change
Expand Up @@ -181,11 +181,10 @@ def one_classifier(df, features, to_predict, names, w_name, w_number, final):
np.set_printoptions(precision=2)

# Plot non-normalized confusion matrix
if True:
plt.figure()
plot_confusion_matrix(
cnf_matrix, classes=names,
title='Confusion matrix, without normalization')
plt.figure()
plot_confusion_matrix(
cnf_matrix, classes=names,
title='Confusion matrix, without normalization')

# Plot normalized confusion matrix
plt.figure()
Expand Down

0 comments on commit 7307016

Please sign in to comment.