search

salesforce / cove

BSD 3-Clause "New" or "Revised" License
473 stars 93 forks source link

How can I train CoVe with my own data？ #4

Closed ttslr closed 6 years ago

ttslr commented 7 years ago

I want to train CoVe with my own train data which contains some sentences without any construction information.

So. How can I make it? Thank you very much!

bmccann commented 6 years ago

I would recommend using something like OpenNMT-py to train an encoder. I branched off that project a long while back to train the pretrained encoder provided by this repo, but you can use any translation implementation online to get CoVe. Once you've trained the model, just save the encoder and use it as you would the pretrained encoder provided by this repo.