search

lxucs / coref-hoi

PyTorch implementation of the end-to-end coreference resolution model with different higher-order inference methods.
Apache License 2.0
59 stars 19 forks source link

Running on our own CoNLL-U files #6

Closed sabyasachee closed 3 years ago

sabyasachee commented 3 years ago

I have been able to run the models on the OntoNotes test set, but how do we get predictions for our own CoNLL-U files?

lxucs commented 3 years ago

Sorry for my late reply. I will add a script soon to perform prediction on custom input.

sabyasachee commented 3 years ago

Thank you. I was able to get predictions on my own CoNLL files but I had to edit the code. It would be great if you could add that functionality, as you said.

lxucs commented 3 years ago

I have just pushed a commit to include a script for prediction on custom input. Let me know if you have more questions.