Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refit a trained model with new triplets #266

Open
aastha19 opened this issue Aug 18, 2022 · 3 comments
Open

Refit a trained model with new triplets #266

aastha19 opened this issue Aug 18, 2022 · 3 comments

Comments

@aastha19
Copy link

aastha19 commented Aug 18, 2022

I have a dataset with new triplets and want to fit this data on an already trained model.
The new dataset is in alliance with the data the model was trained on.
Can this be achieved? If so, how?

@lukostaz
Copy link
Contributor

Unfortunately this is not possible at the moment. I agree it would be a useful feature to open source. Keep an eye on future releases 😉

@sumitpai
Copy link
Contributor

sumitpai commented Aug 18, 2022

You can try this using constant initializer, to initialize your new model with old embeddings. you also need to ensure that model dictionary(model.ent_to_idx, model.rel_to_idx) stays the same with the new data.

However this is not same as continuing training, since the optimizer states wont be the same.

@AbhiPawar5
Copy link

Unfortunately this is not possible at the moment. I agree it would be a useful feature to open source. Keep an eye on future releases 😉

@lukostaz does this mean we have to retrain the model from start if we want to add new triplets? What do you suggest for adding a KG in production that needs to be updated frequently? Thanks.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants