Commit 64d881c8 authored by Marcel Henrik Schubert's avatar Marcel Henrik Schubert
Browse files

fixed axis

parent 08087d91
......@@ -655,10 +655,9 @@ def plotter(subsets, subana, phases, labels):
cnf_matrix = confusion_matrix(df_dic[st][an]['df'][an+'_'+str(st)+'_life_phase'],
life_ph_pred_auth,
labels=phases)
print(cnf_matrix)
print(np.shape(cnf_matrix))
print(len(phases))
f = plt.figure()
ax = f.subplots()
plot_confusion_matrix(cnf_matrix, classes=phases,title=None, ax=ax)
plt.show()
plt.tight_layout()
......@@ -686,7 +685,8 @@ def plotter(subsets, subana, phases, labels):
gen_pred_auth,
labels=['female', 'male'])
f = plt.figure()
plot_confusion_matrix(cnf_matrix, classes=['female', 'male'],title=None)
ax = f.subplots()
plot_confusion_matrix(cnf_matrix, classes=phases,title=None, ax=ax)
plt.tight_layout()
f.savefig(savedir+'heatmaps/cm_{st}_{an}_{label}_{group}.png'.format(st = st, an=an,
label=label,
......@@ -698,6 +698,7 @@ def plotter(subsets, subana, phases, labels):
life_ph_pred_auth_wrong,
labels=phases)
f = plt.figure()
ax = f.subplots()
plot_confusion_matrix(cnf_matrix, classes=phases,title=None, ax=ax)
plt.tight_layout()
f.savefig(savedir+'heatmaps/cm_{st}_{an}_{label}_{group}_false.png'.format(st = st, an=an,
......@@ -709,7 +710,8 @@ def plotter(subsets, subana, phases, labels):
gen_pred_auth_wrong,
labels=['female', 'male'])
f = plt.figure()
plot_confusion_matrix(cnf_matrix, classes=['female', 'male'],title=None, ax=ax)
ax = f.subplots()
plot_confusion_matrix(cnf_matrix, classes=phases,title=None, ax=ax)
plt.tight_layout()
f.savefig(savedir+'heatmaps/cm_{st}_{an}_{label}_{group}_false.png'.format(st = st, an=an,
label=label,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment