benedekrozemberczki/karateclub

save graph2vec model

riyaj8888 opened this issue · 1 comments

riyaj8888 commented

Hi,
How can we save/load the graph2vec model after training?

tomlincr commented

@riyaj8888 you can use the standard techniques for scikit-learn models like pickle, joblib etc.

For example:

from joblib import dump, load
dump(model, 'saved_model.joblib') 
model2 = load('saved_model.joblib') 
embedding = model2.get_embedding()

Reference:

https://scikit-learn.org/stable/model_persistence.html

  • 👍1