Skip to content

Commit be45190

Browse files
authored
Merge pull request #79 from NetherlandsForensicInstitute/fix/eval-compatibility
Update label generation to use `file_name` and `function_name` fields
2 parents db44244 + 70294df commit be45190

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

asmtransformers/scripts/evaluation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def add_label(example):
2222
2323
:param example: row in dataset
2424
:returns example with an extra column containing the label"""
25-
label = example['bin_name'] + '/' + example['func_name']
25+
label = example['file_name'] + '/' + example['function_name']
2626
example['label'] = label
2727
# Ensure same labels are sorted together and in random order.
2828
example['label_random'] = f'{label}\0{random.random()}'

0 commit comments

Comments
 (0)