xu1998hz commited on
Commit
5e0c88b
·
1 Parent(s): b6db393

Update sescore.py

Browse files
Files changed (1) hide show
  1. sescore.py +0 -2
sescore.py CHANGED
@@ -145,8 +145,6 @@ class SEScore(evaluate.Metric):
145
  gpus = 1 if torch.cuda.is_available() else 0
146
 
147
  data = {"src": references, "mt": predictions}
148
- print(data)
149
  data = [dict(zip(data, t)) for t in zip(*data.values())]
150
- print(data)
151
  scores, mean_score = self.scorer.predict(data, gpus=gpus, progress_bar=progress_bar)
152
  return {"mean_score": mean_score, "scores": scores}
 
145
  gpus = 1 if torch.cuda.is_available() else 0
146
 
147
  data = {"src": references, "mt": predictions}
 
148
  data = [dict(zip(data, t)) for t in zip(*data.values())]
 
149
  scores, mean_score = self.scorer.predict(data, gpus=gpus, progress_bar=progress_bar)
150
  return {"mean_score": mean_score, "scores": scores}