Merge pull request #172 from cognitive-catalyst/sue-branch

Fix warning on confusion matrix DCO 1.1 Signed-off-by: Sue S Schmidt sueschm@us.ibm.com

Author: andrewrfreed

utils/confusionmatrix.py

@@ -54,10 +54,11 @@ def func(args):
     #Plot a normalized confusion matrix as a heatmap
     plt.figure(figsize = (10,10))
     df_cm = pd.DataFrame(output_matrix, index=index_labels, columns=column_labels)
+    df_cm = df_cm.to_numpy()
     df_cm = df_cm.astype('float') / df_cm.sum(axis=1)[:, np.newaxis]
     sns.set(font_scale=1)
     # Add 'annot=True' to the list of options for heatmap if you want to print the numbers, ideal only for small maps
-    hm = sns.heatmap(df_cm, cmap="Greys",cbar=False,fmt='.1%',linewidths=0.1,linecolor='black')
+    hm = sns.heatmap(df_cm, cmap="Greys",cbar=False,fmt='.1%',linewidths=0.1,linecolor='black',xticklabels=column_labels, yticklabels=index_labels)
     hm.set_yticklabels(hm.get_yticklabels(), rotation=0)
     hm.set_xticklabels(hm.get_xticklabels(), rotation=90)
     plt.title("Normalized Confusion Matrix")