Fix get_generated_reports missing parameter

Author: ttanida

src/full_model/evaluate_full_model/evaluate_language_model.py

@@ -1178,6 +1178,7 @@ def evaluate_language_model(model, val_dl, tokenizer, writer, run_params, genera
 
     # used in function get_generated_reports
     sentence_tokenizer = spacy.load("en_core_web_trf")
+    bert_score = evaluate.load("bertscore")
 
     with torch.no_grad():
         for num_batch, batch in tqdm(enumerate(val_dl), total=NUM_BATCHES_TO_PROCESS_FOR_LANGUAGE_MODEL_EVALUATION):
@@ -1255,7 +1256,8 @@ def evaluate_language_model(model, val_dl, tokenizer, writer, run_params, genera
                 generated_sents_for_selected_regions,
                 selected_regions,
                 sentence_tokenizer,
-                BERTSCORE_SIMILARITY_THRESHOLD
+                BERTSCORE_SIMILARITY_THRESHOLD,
+                bert_score,
             )
 
             gen_and_ref_sentences["generated_sentences"].extend(generated_sents_for_selected_regions)

src/full_model/test_set_evaluation.py

@@ -343,15 +343,8 @@ def iterate_over_test_loader(test_loader):
                     generated_sents_for_selected_regions,
                     selected_regions,
                     sentence_tokenizer,
-                    BERTSCORE_SIMILARITY_THRESHOLD
-                generated_reports, removed_similar_generated_sentences = (
-                    get_generated_reports(
-                        generated_sents_for_selected_regions,
-                        selected_regions,
-                        sentence_tokenizer,
-                        BERTSCORE_SIMILARITY_THRESHOLD,
-                        bert_score,
-                    )
+                    BERTSCORE_SIMILARITY_THRESHOLD,
+                    bert_score,
                 )
 
                 gen_and_ref_sentences["generated_sentences"].extend(generated_sents_for_selected_regions)