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

updated logic for the extraction of correct

Browse Source 
predictions
This commit is contained in:
Karma Riuk
2025-06-10 23:41:45 +02:00
parent f36fcc6e05
commit 21de0ffd7a
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
extract_correct_predictions.py
View File

@ -4,6 +4,7 @@ import os, json, tarfile, argparse
from typing import Optional from typing import Optional
from dataset import Dataset, ArchiveState from dataset import Dataset, ArchiveState
from utils import EnumChoicesAction from utils import EnumChoicesAction
from tqdm import tqdm
class OutputType(Enum): class OutputType(Enum):
@ -69,9 +70,9 @@ def extract_refinement_predictions(dataset_path: str, archives_path: str, output
results = {} results = {}
# Iterate over entries that address the change # Iterate over entries that address the change
for entry in dataset.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 sel.is_code_related): if not sel or not (sel.diff_after_address_change and entry.metadata.is_code_related):
continue continue
entry_id = entry.metadata.id entry_id = entry.metadata.id