зеркало из https://github.com/microsoft/IRNet.git
Fix bug in measurement
This commit is contained in:
Родитель
3f8224514c
Коммит
72df5c876f
6
train.py
6
train.py
|
@ -103,11 +103,11 @@ def train(args):
|
|||
print(tb)
|
||||
else:
|
||||
utils.save_checkpoint(model, os.path.join(model_save_path, 'end_model.model'))
|
||||
json_datas = utils.epoch_acc(model, args.batch_size, val_sql_data, val_table_data,
|
||||
json_datas, sketch_acc, acc = utils.epoch_acc(model, args.batch_size, val_sql_data, val_table_data,
|
||||
beam_size=args.beam_size)
|
||||
acc = utils.eval_acc(json_datas, val_sql_data)
|
||||
# acc = utils.eval_acc(json_datas, val_sql_data)
|
||||
|
||||
print("Sketch Acc: %f, Acc: %f, Beam Acc: %f" % (acc, acc, acc,))
|
||||
print("Sketch Acc: %f, Acc: %f, Beam Acc: %f" % (sketch_acc, acc, acc,))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
Загрузка…
Ссылка в новой задаче