search

memray / seq2seq-keyphrase

MIT License
318 stars 109 forks source link

Transfer raw data to the same format in baseline-data #9

Closed lostnighter closed 6 years ago

lostnighter commented 6 years ago

I want to extract keyphrases from my own data using the model, but how to transfer raw data to the same format in baseline-data? what is the processing flow or call some func?

memray commented 6 years ago

The line 518 of keyphrase_copynet.py will load the external dataset according to the given dataset_name. I did implement some loading func for a few datasets and you can customize a loader for your own data. Basically there is just one function get_docs() needs implementing.