如何在多索引数据帧上创建混淆矩阵

时间:2018-07-18 12:42:40

标签: python multi-index hierarchical confusion-matrix

基于下面的数据框格式

table

我应该如何创建混淆矩阵?在momoment上,我收到以下错误:

ValueError: You appear to be using a legacy multi-label data representation. Sequence of sequences are no longer supported; use a binary array or sparse matrix instead.

当我将其用于1级数据帧时它可以工作,但是由于这是分层的,因此出现上述错误。我怀疑这与我调用专栏的方式有关。

我的代码如下(标签不完整,但仅作为示例):

sns.set(style='whitegrid', rc={"grid.linewidth": 0.1})
sns.set_context("paper", font_scale=2)                                                  
plt.figure(figsize=(20, 20))                                                                                                                                                                                                            
color = sns.color_palette("Set1", 6)

cm = confusion_matrix(meta['car','type'],meta['bike','type'],labels)

ax= plt.subplot()
sns.heatmap(cm, annot=True, ax = ax,annot_kws=  {"size":15},fmt="d",cmap='Blues'); #annot=True to annotate cells

# labels, title and ticks
ax.set_xlabel('car',va='top');
ax.set_ylabel('car'); 

ax.set_title('bike\n', loc='center'); 
ax.xaxis.set_ticklabels(["yellow","big","blue","black"],va='center');
ax.xaxis.set_ticks_position('top')

0 个答案:

没有答案