Commit 08087d91 authored by Marcel Henrik Schubert's avatar Marcel Henrik Schubert
Browse files

error finding

parent 1ed13215
......@@ -40,8 +40,10 @@ savedir = '/draco/ptmp/mschuber/PAN/Data/results/'
#path = '../Data/pan19-celebrity-profiling-training-dataset-2019-01-31/stratified_subsample/'
subana_l = ['org/', 'min_tweets_1000/', 'complete_balance/']
subana_l = ['org/']
subsets_l = [200, 500, 1000, 2000]
subsets_l = [200]
classifiers = ['SVM']
......@@ -612,7 +614,7 @@ def plotter(subsets, subana, phases, labels):
an=an,
label=label,
group='life_phase'))
plt.close()
......@@ -653,8 +655,12 @@ 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()
plot_confusion_matrix(cnf_matrix, classes=phases,title=None, ax=ax)
plt.show()
plt.tight_layout()
f.savefig(savedir+'heatmaps/cm_{st}_{an}_{label}_{group}.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