graykode / commit-autosuggestions

Go to a project
Toggle navigation Toggle navigation pinning
graykode
26618a00 1 parent ad248582

(fixed) type of gold.target change from list to str

Inline Side-by-side
Showing 1 changed file with 2 additions and 2 deletions
......@@ -466,7 +466,7 @@ def main():
for ref,gold in zip(p,eval_examples):
predictions.append(str(gold.idx)+'\t'+ref)
f.write(str(gold.idx)+'\t'+ref+'\n')
f1.write(str(gold.idx)+'\t'+gold.target+'\n')
f1.write(str(gold.idx)+'\t'+' '.join(gold.target)+'\n')
(goldMap, predictionMap) = bleu.computeMaps(predictions, os.path.join(args.output_dir, "dev.gold"))
dev_bleu=round(bleu.bleuFromMaps(goldMap, predictionMap)[0],2)
......@@ -523,7 +523,7 @@ def main():
for ref,gold in zip(p,eval_examples):
predictions.append(str(gold.idx)+'\t'+ref)
f.write(str(gold.idx)+'\t'+ref+'\n')
f1.write(str(gold.idx)+'\t'+gold.target+'\n')
f1.write(str(gold.idx)+'\t'+' '.join(gold.target)+'\n')
(goldMap, predictionMap) = bleu.computeMaps(predictions, os.path.join(args.output_dir, "test_{}.gold".format(idx)))
dev_bleu=round(bleu.bleuFromMaps(goldMap, predictionMap)[0],2)
......