This repository contains a implementation of our "WalkLM: A Uniform Language Model Fine-tuning Framework for Attributed Graph Embedding" .

1. Pytorch 1.12.1
2. Python 3.7.15

The implementation of embedding generate (emb.py)、node classification task(nc.py) and link prediction task (lp.py);

python emb.py

python nc.py

python lp.py

If you find the code useful, please consider citing the following paper:

@inproceedings{tan2023walklm,
  title={WalkLM: A Uniform Language Model Fine-tuning Framework for Attributed Graph Embedding},
  author={Tan, Yanchao and Zhou, Zihao and Lv, Hang and Liu, Weiming and Yang, Carl},
  booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
  year={2023}
}