源代码:
将matplotlib.pyplot导入为情节
从sklearn.metrics导入confusion_matrix
将seaborn导入为sns
将熊猫作为pd导入
plot.style.use('ggplot')
cm = confusion_matrix(y_test,y_pred)
df_cm = pd.DataFrame(cm,索引=(0,1),列=(0,1))
plot.figure(figsize =(8,7))
sns.set(font_scale = 1.4)
sns.heatmap(df_cm,annot = True,fmt ='g')
从代码的执行中,我获得了显示在几乎不可见的单元格边缘上的数据的标签值。我正在使用jupyter笔记本。