77755d613b027b2c5a80297cca2deb74130b4b1f,t2t_csaky/scripts/hred_metrics.py,,main,#,351

Before Change

  //m.metrics()
  //m.write_metrics()
  for i in [22001, 33001, 44001, 66001, 88001, 109001, 142001, 175001, 218001, 225001]:
    m = Metrics("decode_dir/DailyDialog/trf_20_dropout-base/dev_set_" + str(i) + ".txt")
    m.metrics()
    m.write_metrics()


if __name__ == "__main__":
  main()

After Change

def main():
  m = Metrics(sys.argv[1]) if len(sys.argv) > 1 else Metrics()
  m.metrics()
  m.write_metrics()
  
  folder = "decode_dir/DailyDialog/temp_overfit/"
  for file_name in os.listdir(folder):
    m = Metrics(folder + file_name)

Instances