Removed Debug Code and Prints

isaac
Sai Pavan Mandava 5 years ago
parent 6946077d94
commit 64f787b5d7

@ -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")
y_pred, y_true = pred.load_model_and_predict_test_data("/mount/arbeitsdaten/studenten1/team-lab-nlp/mandavsi_rileyic/saved_models/experiment_4")

@ -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

Loading…
Cancel
Save