snsd0805/vlnbert
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: change 'swap' to 'found'

Browse Source
This commit is contained in:
Ting-Jun Wang 2023-11-06 18:30:31 +08:00
parent a5db597de5
commit 4936098b5e
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
adversarial_summary.py
View File

@ -19,13 +19,13 @@ for file in ['train', 'val_unseen', 'val_seen', 'train_seen', 'test', 'val_train
'path': i['path'], 'path': i['path'],
'heading': float(i['heading']), 'heading': float(i['heading']),
'instructions': [ i['instruction'] ], 'instructions': [ i['instruction'] ],
'swap': [ True if i['found'] else False ], 'found': [ i['found'] ],
'id': i['id'], 'id': i['id'],
'objId': i['objId'] 'objId': i['objId']
} }
else: else:
result[instruction_id]['instructions'].append(i['instruction']) result[instruction_id]['instructions'].append(i['instruction'])
result[instruction_id]['swap'].append( True if i['found'] else False) result[instruction_id]['found'].append( i['found'] )
output = [] output = []
for k, item in result.items(): for k, item in result.items():