AlienChen commited on
Commit
90d840d
·
verified ·
1 Parent(s): 41d0b8c

Update moppit.py

Browse files
Files changed (1) hide show
  1. moppit.py +5 -1
moppit.py CHANGED
@@ -147,7 +147,11 @@ for i in range(args.n_batches):
147
  scores.append(score.item())
148
  else:
149
  scores.append(candidate_scores.item())
150
- print(scores)
 
 
 
 
151
 
152
  with open(args.output_file, 'a') as f:
153
  f.write(input_seqs[0])
 
147
  scores.append(score.item())
148
  else:
149
  scores.append(candidate_scores.item())
150
+
151
+ print(f"Sample: {input_seqs[0]")
152
+ print(f"Scores: ")
153
+ for i, objective in args.objectives:
154
+ print(f"{objective}: {scores[i]:.4f}")
155
 
156
  with open(args.output_file, 'a') as f:
157
  f.write(input_seqs[0])