/line

TensorFlow implementation of paper "LINE: Large-scale Information Network Embedding" by Jian Tang, et al.

Primary LanguagePython

LINE in TensorFlow

TensorFlow implementation of paper LINE: Large-scale Information Network Embedding by Jian Tang, et al.

You can see my slide on GDG DevFest 2017 for more detail about LINE and TensorFlow. Notice: code shown in the slide are pseudocode, minibatch and negative sampling are omitted in the slide.

Prerequisites

  • Python 3.6
  • TensorFlow 1.3.0
  • Networkx
  • NumPy

Setup

  • Prepare a network using networkx. Write the graph to a file by nx.write_gpickle.
  • Put the network file in data folder.
  • Run line.py --graph_file graph.pkl to start training. graph.pkl is the name of your network file.
  • Embedding will be stored in data/embedding_XXX-order.pkl. You can load it by pickle.load() in python.

References