search

memray / OpenNMT-kpg-release

Keyphrase Generation
MIT License
216 stars 34 forks source link

Problem with generating kephrases (inference.ipynb file) #39

Closed mmjaz closed 3 years ago

mmjaz commented 3 years ago

Hello I have a problem with generating keyphrases for single text

Traceback (most recent call last):
  File "/home/***/PycharmProjects/OpenNMT-kpg-release/notebook/inference.py", line 69, in <module>
    opt=opt
  File "/home/***/PycharmProjects/OpenNMT-kpg-release/onmt/translate/translator.py", line 396, in translate
    for batch in data_iter:
  File "/home/***/PycharmProjects/OpenNMT-kpg-release/onmt/inputters/inputter.py", line 972, in __iter__
    self.init_epoch()
  File "/home/***/anaconda3/envs/torch17/lib/python3.7/site-packages/torchtext/data/iterator.py", line 117, in init_epoch
    self.create_batches()
  File "/home/***/PycharmProjects/OpenNMT-kpg-release/onmt/inputters/inputter.py", line 960, in create_batches
    b = keyphrase_dataset.process_multiple_tgts(b, self.dataset.tgt_type)
  File "/home/***/PycharmProjects/OpenNMT-kpg-release/onmt/inputters/keyphrase_dataset.py", line 370, in process_multiple_tgts
    order = obtain_sorted_indices(ex.src, ex.tgt, sort_by=tgt_type)
AttributeError: 'Example' object has no attribute 'tgt'

Process finished with exit code 1