IITH-Compilers / IR2Vec

Implementation of IR2Vec, LLVM IR Based Scalable Program Embeddings
Other
84 stars 39 forks source link

OpenKE Pytorch compatibility with link prediction scores #119

Closed Charanyash closed 2 months ago

svkeerthy commented 3 months ago

Please fix the formatting to address pre-commit check failure. You may need to install the pre-commit package and setup the hooks given here.

Charanyash commented 3 months ago

I have updated the code with requested changes and added flags for link prediction and analogies calculations while training. Please check the latest code