CoderPat / structured-neural-summarization

A repository with the code for the paper with the same title
MIT License
74 stars 26 forks source link

Bug in inference code #12

Closed ioana-blue closed 5 years ago

ioana-blue commented 5 years ago

Starting from observations from #10 I think I found a bug in infer function:

https://github.com/CoderPat/structured-neural-summarization/blob/master/train_and_eval.py#L419

Look at the parameters that are passed:

        features_file=args.train_source_file,
        labels_file=args.train_target_file,

I think those files should be the inference files not the train files. This code does inference on train files. That's why I was getting thousands of outputs for an inference file with 2 samples.