plot path print

5 years ago · bd559e0d8a
parent 87efce8f82
commit bd559e0d8a
1 changed files with 4 additions and 1 deletions
--- 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)