From 64f787b5d7635696fae5f64e2c9ce9bffee1cdfb Mon Sep 17 00:00:00 2001 From: Sai Pavan Mandava Date: Wed, 29 Jul 2020 19:56:28 +0200 Subject: [PATCH] Removed Debug Code and Prints --- run.py | 2 +- testing/intent_predictor.py | 11 ++--------- 2 files changed, 3 insertions(+), 10 deletions(-) diff --git a/run.py b/run.py index 8e616e0..f80a37b 100644 --- a/run.py +++ b/run.py @@ -1,4 +1,4 @@ import classifier import testing.intent_predictor as pred -pred.load_model_and_run_predictions("/mount/arbeitsdaten/studenten1/team-lab-nlp/mandavsi_rileyic/saved_models/experiment_4") \ No newline at end of file +y_pred, y_true = pred.load_model_and_predict_test_data("/mount/arbeitsdaten/studenten1/team-lab-nlp/mandavsi_rileyic/saved_models/experiment_4") diff --git a/testing/intent_predictor.py b/testing/intent_predictor.py index f7574cb..4e00735 100644 --- a/testing/intent_predictor.py +++ b/testing/intent_predictor.py @@ -36,21 +36,15 @@ def make_predictions(model: Model, dataset_reader: DatasetReader, file_path: str vocab = model.vocab jsonl_reader = DataReaderJsonLines(file_path) - i = 0 for citation in jsonl_reader.read(): - i += 1 true_list.append(citation.intent) output = predictor.predict(citation.text, citation.intent) prediction_list.append(output['prediction']) - # prediction_list.append({vocab.get_token_from_index(label_id, 'labels'): prob - # for label_id, prob in enumerate(output['probabilities'])}) - if i == 10: - break return prediction_list, true_list -def load_model_and_run_predictions(saved_model_dir: str): +def load_model_and_predict_test_data(saved_model_dir: str): project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) dev_file_path = project_root + '/data/jsonl/dev.jsonl' test_file_path = project_root + '/data/jsonl/test.jsonl' @@ -60,5 +54,4 @@ def load_model_and_run_predictions(saved_model_dir: str): y_pred, y_true = make_predictions(model_archive.model, citation_dataset_reader, test_file_path) - print('Predictions ', y_pred) - print('True Labels ', y_true) + retun y_pred,y_true