From d18593f869c3048b917145c388e6ab28e5e9b2e6 Mon Sep 17 00:00:00 2001 From: Pavan Mandava Date: Wed, 29 Jul 2020 20:29:59 +0200 Subject: [PATCH] printing confusion matrix --- run.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/run.py b/run.py index b002a47..1cbccd2 100644 --- a/run.py +++ b/run.py @@ -5,4 +5,8 @@ import eval.metrics as metrics y_pred, y_true = pred.load_model_and_predict_test_data("/mount/arbeitsdaten/studenten1/team-lab-nlp/mandavsi_rileyic/saved_models/experiment_4") -metrics.plot_confusion_matrix(metrics.get_confusion_matrix(y_true, y_pred), "BiLSTM Classifier + Attention with ELMo") \ No newline at end of file +confusion_matrix = metrics.get_confusion_matrix(y_true, y_pred) + +print(confusion_matrix) + +metrics.plot_confusion_matrix(confusion_matrix, "BiLSTM Classifier + Attention with ELMo")