GraphZoom is a framework that aims to improve both performance and scalability of graph embedding techniques.
- Matlab Compiler Runtime (MCR) 2018a(Linux), which is a standalone set of shared libraries that enables the execution of compiled MATLAB applications and does not require license to install.
- numpy
- networkx
- scipy
- scikit-learn
- gensim, only required by deepwalk, node2vec
- theano, only required by netmf
- tensorflow, only required by graphsage
- install matlab compiler runtime 2018a (https://www.mathworks.com/products/compiler/matlab-runtime.html)
- pip install -r requirements.txt
Note: You have to pass the root directory of matlab compiler runtime to the argument "--mcr_dir" when running graphzoom.py
- Example Usage
cd graphzoom
python graphzoom.py --mcr_dir YOUR_MCR_ROOT_DIR --dataset citeseer --search_ratio 12 --num_neighs 10 --embed_method deepwalk
--mcr_dir: root directory of matlab compiler runtime
--dataset: input dataset, currently supports "json" format
--embed_method: choose a specific basic embedding algorithm
--search_ratio: control the search space of graph fusion
--num_neighs: control number of edges in feature graph
- Full Command List
- The full list of command line options is available with
python graphzoom.py --help