karma/crab
1
0
Fork 0
mirror of https://github.com/karma-riuk/crab.git synced 2025-07-05 13:38:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

fixed extraction of correct predictions

Browse Source
This commit is contained in:
Karma Riuk
2025-06-12 13:25:41 +02:00
parent 21de0ffd7a
commit f1204c4c35
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
extract_correct_predictions.py
View File

@ -72,7 +72,11 @@ def extract_refinement_predictions(dataset_path: str, archives_path: str, output
# Iterate over entries that address the change # Iterate over entries that address the change
for entry in tqdm(dataset.entries, unit="entries"): for entry in tqdm(dataset.entries, unit="entries"):
sel = entry.metadata.selection sel = entry.metadata.selection
if not sel or not (sel.diff_after_address_change and entry.metadata.is_code_related): if not sel or not (
sel.diff_after_address_change
and entry.metadata.is_code_related
and entry.metadata.is_covered
):
continue continue
entry_id = entry.metadata.id entry_id = entry.metadata.id