KeyError: 'NewsDNA__argument_labels'

dwadden / dygiepp

Span-based system for named entity, relation, and event extraction.

MIT License

575 stars 120 forks source link

KeyError: 'NewsDNA__argument_labels' #54

Closed desothier1 closed 3 years ago

desothier1 commented 3 years ago

I tried to train and test with a new dataset that I formatted similar to ace event data

However, I keep having the next error

File "/home/thierry/repos/dygiepp-dev/dygie/models/events.py", line 177, in forward mention_pruner = self._mention_pruners[self._active_namespaces["argument"]] File "/home/thierry/miniconda3/lib/python3.8/site-packages/torch/nn/modules/container.py", line 286, in getitem return self._modules[key]

I assume I should specify the used labels somewhere ?

Thanks for helping

dwadden commented 3 years ago

Hi, thanks for your interest.

Can you confirm that your dataset conforms to the requirements outlined in https://github.com/dwadden/dygiepp/blob/master/doc/data.md?
If so, can you post a full stack trace showing the error you're getting?

I'll look at the full stack trace. If we're still stuck, you can send me a couple instances from your dataset and I can try to run the code myself.

dwadden commented 3 years ago

Closing due to lack of activity. Feel free to re-open if you're still stuck.

desothier1 commented 3 years ago

Thanks for your comments and help. There was indeed a data format error that I could fix based on the requirements info.

dwadden commented 3 years ago

OK, glad it worked out.