YYama0 commited on
Commit
a65dbaf
·
verified ·
1 Parent(s): ffa434e

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -264,7 +264,7 @@ def main():
264
  if args.binary:
265
  suitable_count = results['suitable_for_vqa'].sum()
266
  else:
267
- suitable_count = (results['prediction'] > 0.5).sum()
268
 
269
  total_count = len(results)
270
 
 
264
  if args.binary:
265
  suitable_count = results['suitable_for_vqa'].sum()
266
  else:
267
+ suitable_count = (results['prediction'] > 0.9).sum()
268
 
269
  total_count = len(results)
270