I have a use-case to train graph embeddings, looking for a way to do it in pytorch and tensorflow. The restriction is the methodology should incorporate edge weights in calculating transition probability. Can someone please advise the best framework/library to do. The node2vec implementation by the authors is in spark only, however my current ML pipeline is more pytorch/TF friendly.



Don't know if this is what you need but I know of the Ampligraph library:

Python library for Representation Learning on Knowledge Graphs https://docs.ampligraph.org

| improve this answer | |

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.