Merge pull request #894 from rasbt/class-names

fix conf mat for string class names

Author: rasbt

docs/sources/CHANGELOG.md

@@ -35,6 +35,7 @@ The CHANGELOG for the current development version is available at
 ##### Bug Fixes
 
 - Fix unreadable labels in `heatmap` for certain colormaps. ([#852](https://github.com/rasbt/mlxtend/pull/852))
+- Fix an issue in `mlxtend.plotting.plot_confusion_matrix` when string class names are passed ([#894](https://github.com/rasbt/mlxtend/pull/894))
 
 
 

mlxtend/plotting/plot_confusion_matrix.py

@@ -120,7 +120,8 @@ def plot_confusion_matrix(conf_mat,
         for j in range(conf_mat.shape[1]):
             cell_text = ""
             if show_absolute:
-                cell_text += format(conf_mat[i, j], 'd')
+                num = conf_mat[i, j].astype(np.int64) 
+                cell_text += format(num, 'd')
                 if show_normed:
                     cell_text += "\n" + '('
                     cell_text += format(normed_conf_mat[i, j], '.2f') + ')'