goleey/alp-baselines

Baselines for anchor link prediction (including MNA, PALE, FINAL, FRUI-P)

MNA

Citation: Kong, X., Zhang, J., & Yu, P. S. (2013). Inferring anchor links across multiple heterogeneous social networks. In Proceedings of the 22nd ACM international conference on Conference on information & knowledge management - CIKM ’13 (pp. 179–188).

python alp_main.py --graph1 data/test.src.net --graph2 data/test.obj.net --identity-linkage data/test.align --output test.res --method mna

FINAL

Citation: Zhang, S. (2016). FINAL: Fast Attributed Network Alignment. In Proceedings of the 22th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining - KDD '16 (pp. 421–434).

python alp_main.py --method final --graph1 data/test.src.net --graph2 data/test.obj.net --graph-size1 20 --graph-size2 20 --identity-linkage data/test.align --output test

FRUI-P

Citation: Zhou, X., Liang, X., Member, S., Du, X., & Zhao, J. (2018). Structure Based User Identification across Social Networks. IEEE Transactions on Knowledge and Data Engineering, 30(6), 1178–1191.

• Step 1: network embedding (FFVM)

python ne_main.py --input data/test.src.net --output ffvm.res --batch-size 6 --table-size 100 --rep-size 4 --method ffvmx --neg-ratio 2 --order 1

• Step 2: matching (FRUI-P)

python alp_main.py --embedding1 test.res.epoch5.node_order1 --embedding2 ffvm.res.epoch5.node --graph1 data/test.src.net --graph2 data/test.obj.net --identity-linkage data/test.align --output test.fruip --epochs 10 --method fruip

PALE

Citation: Man, T., Shen, H., Liu, S., Jin, X., & Cheng, X.. Predict anchor links across social networks via an embedding approach. IJCAI International Joint Conference on Artificial Intelligence, 2016, 1823–1829.

• Step 1: network embedding (LINE)

python ne_main.py --input data/test.src.net --output test.res --batch-size 6 --table-size 100 --rep-size 4 --method line --neg-ratio 2 --order 1

• Step 2: matching (PALE)

python alp_main.py --embedding1 test.res.epoch5.node_order1 --embedding2 ffvm.res.epoch5.node --type-model lin --identity-linkage data/test.align --output test.alp --batch-size 4 --input-size 4 --epochs 10 --method pale --neg-ratio 5 --device :/cpu:0