diff --git a/predict.py b/predict.py index c5c3a37..394189c 100644 --- a/predict.py +++ b/predict.py @@ -10,4 +10,7 @@ confusion_matrix = metrics.get_confusion_matrix(y_true, y_pred) print(confusion_matrix) -metrics.plot_confusion_matrix(confusion_matrix, "BiLSTM Classifier + Attention with ELMo", model_path+'/confusion_matrix_plot.png') +plot_file_path = model_path+'/confusion_matrix_plot.png' +metrics.plot_confusion_matrix(confusion_matrix, "BiLSTM Classifier + Attention with ELMo", plot_file_path) + +print('Confusion Matrix Plot saved to :: ', plot_file_path)