Skip to content
Snippets Groups Projects
Commit d8f05949 authored by Hsien-Chin Lin's avatar Hsien-Chin Lin
Browse files

wip

parent 40520ae5
No related branches found
No related tags found
No related merge requests found
...@@ -145,6 +145,7 @@ class Evaluator: ...@@ -145,6 +145,7 @@ class Evaluator:
json.dump(nlg_eval, json.dump(nlg_eval,
open(os.path.join(dir_name, "nlg_eval.json"), 'w'), open(os.path.join(dir_name, "nlg_eval.json"), 'w'),
indent=2) indent=2)
return os.path.join(dir_name, "nlg_eval.json")
def evaluation(self, input_file=None, generated_file=None): def evaluation(self, input_file=None, generated_file=None):
force_prediction = True force_prediction = True
...@@ -241,10 +242,15 @@ def main(): ...@@ -241,10 +242,15 @@ def main():
if args.do_semantic: if args.do_semantic:
eval.evaluation(args.input_file) eval.evaluation(args.input_file)
if args.do_nlg: if args.do_nlg:
eval.nlg_evaluation(input_file=args.input_file, nlg_result = eval.nlg_evaluation(input_file=args.input_file,
generated_file=args.generated_file, generated_file=args.generated_file,
golden=args.do_golden_nlg) golden=args.do_golden_nlg)
eval.evaluation(args.input_file) if args.generated_file:
generated_file = args.generated_file
else:
generated_file = nlg_result
eval.evaluation(args.input_file,
generated_file)
if __name__ == '__main__': if __name__ == '__main__':
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment