karma/crab
1
0
Fork 0
mirror of https://github.com/karma-riuk/crab.git synced 2025-07-24 12:57:48 +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 dataset import Dataset, ArchiveState
from utils import EnumChoicesAction
from tqdm import tqdm
class OutputType(Enum):
@@ -69,9 +70,9 @@ def extract_refinement_predictions(dataset_path: str, archives_path: str, output
results = {}
# Iterate over entries that address the change
for entry in dataset.entries:
for entry in tqdm(dataset.entries, unit="entries"):
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
entry_id = entry.metadata.id