You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
576 B
17 lines
576 B
import classifier
|
|
import testing.intent_predictor as pred
|
|
|
|
import eval.metrics as metrics
|
|
|
|
model_path = '/mount/arbeitsdaten/studenten1/team-lab-nlp/mandavsi_rileyic/saved_models/experiment_4'
|
|
y_pred, y_true = pred.load_model_and_predict_test_data(model_path)
|
|
|
|
confusion_matrix = metrics.get_confusion_matrix(y_true, y_pred)
|
|
|
|
print(confusion_matrix)
|
|
|
|
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)
|