Skip to content

This repo contains a python implementation of an algorithm which is used on the link prediction on the attributed network.

License

Notifications You must be signed in to change notification settings

MetaronWang/graph-embedding-with-attribute

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

It's just a messy version so far, which is the initial one. I will refactor this project later on.

Dependence

  • Tensorflow=1.14
  • Numpy=1.19.2
  • gensim=3.8.3
  • sklearn=0.23.2

How to Run

You just need to run the scripts below in the project root folder.

If you don't want to run the MLP classifier, you don't need to install the TensorFlow package.

Do the pre-training

python node2vec_walk.py

Do the classification

For SVM、Logistic Regression、NaiveBayes、Decision Tree、Random Forest

python node_classifier.py

For MLP

python nn_classifier.py

About

This repo contains a python implementation of an algorithm which is used on the link prediction on the attributed network.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages