Я хочу создать визуализацию матрицы путаницы, используя matplotlib. Параметры нижеуказанных методов - это метки классов (алфавит), результаты классификации являются списком списков (conf_arr) и выходным именем файла. Я доволен результатом до сих пор, с одной последней проблемой:
Я не могу центрировать метки тика оси между линиями сетки. Если передать параметр степени в метод imshow, как показано ниже, сетка выравнивается так, как мне бы хотелось. Если я прокомментирую это, сетка будет пропущена, но наклейки - это то, что я хотел бы они должны быть. Я думаю, мне нужен способ переместить метку метки между связанным тиком и следующим тиком но я не знаю, возможно ли и как это возможно.
Подводя итог, я хочу, чтобы сетка/тики была похожа на левое изображение, но метки метки позиционируется как в правильном изображении:
def create_confusion_matrix(alphabet, conf_arr, outputname):
norm_conf = []
width = len(conf_arr)
height = len(conf_arr[0])
for i in conf_arr:
a = 0
tmp_arr = []
a = sum(i, 0)
for j in i:
tmp_arr.append(float(j)/float(a))
norm_conf.append(tmp_arr)
fig = plt.figure(figsize=(14,14))
#fig = plt.figure()
plt.clf()
ax = fig.add_subplot(111)
ax.set_aspect(1)
ax.grid(which='major')
res = ax.imshow(np.array(norm_conf), cmap=plt.cm.binary,
interpolation='none', aspect='1', vmax=1,
##Commenting out this line sets labels correctly,
##but the grid is off
extent=[0, width, height, 0]
)
divider = make_axes_locatable(ax)
cax = divider.append_axes("right", size="5%", pad=0.2)
cb = fig.colorbar(res, cax=cax)
#Axes
ax.set_xticks(range(width))
ax.set_xticklabels(alphabet, rotation='vertical')
ax.xaxis.labelpad = 0.5
ax.set_yticks(range(height))
ax.set_yticklabels(alphabet, rotation='horizontal')
#plt.tight_layout()
plt.savefig(outputname, format='png')
Полученное изображение выглядит следующим образом: