saefro991 commited on
Commit
df84cff
1 Parent(s): e13d732

fix csv output

Browse files
Files changed (1) hide show
  1. predict.py +3 -2
predict.py CHANGED
@@ -77,8 +77,9 @@ def main():
77
  pass
78
  for batch in tqdm.tqdm(loader):
79
  scores = scorer.score(batch.to(device))
80
- with open(args.out_path, 'a') as fw:
81
- fw.write("\n".join([str(s) for s in scores]))
 
82
 
83
 
84
  if __name__ == '__main__':
77
  pass
78
  for batch in tqdm.tqdm(loader):
79
  scores = scorer.score(batch.to(device))
80
+ with open(args.out_path, 'a') as fw:
81
+ for s in scores:
82
+ fw.write(str(s) + "\n")
83
 
84
 
85
  if __name__ == '__main__':