Giter Club home page Giter Club logo

drne's Introduction

DRNE

The Implementation of "Deep Recursive Network Embedding with Regular Equivalence"(KDD 2018).

Requirements

Python >= 3.5.2
scipy >= 0.19.1
numpy >= 1.13.1
tensorflow == 1.2.0
networkx >= 1.11

Usage

Example Usage
python src/main.py --data_path dataset/barbell.edgelist --save_path result/barbell --save_suffix test \
      -s 16 -b 256 -lr 0.0025 --index_from_0 True
Full Command List
usage: Deep Recursive Network Embedding with Regular Equivalence
       [-h] [--data_path DATA_PATH] [--save_path SAVE_PATH]
       [--save_suffix SAVE_SUFFIX] [-s EMBEDDING_SIZE] [-e EPOCHS_TO_TRAIN]
       [-b BATCH_SIZE] [-lr LEARNING_RATE] [--undirected UNDIRECTED]
       [-a ALPHA] [-l LAMB] [-g GRAD_CLIP] [-K K]
       [--sampling_size SAMPLING_SIZE] [--seed SEED]
       [--index_from_0 INDEX_FROM_0]

optional arguments:
  -h, --help            show this help message and exit
  --data_path DATA_PATH
                        Directory to load data.
  --save_path SAVE_PATH
                        Directory to save data.
  --save_suffix SAVE_SUFFIX
                        Directory to save data.
  -s EMBEDDING_SIZE, --embedding_size EMBEDDING_SIZE
                        the embedding dimension size
  -e EPOCHS_TO_TRAIN, --epochs_to_train EPOCHS_TO_TRAIN
                        Number of epoch to train. Each epoch processes the
                        training data once completely
  -b BATCH_SIZE, --batch_size BATCH_SIZE
                        Number of training examples processed per step
  -lr LEARNING_RATE, --learning_rate LEARNING_RATE
                        initial learning rate
  --undirected UNDIRECTED
                        whether it is an undirected graph
  -a ALPHA, --alpha ALPHA
                        the rate of structure loss and orth loss
  -l LAMB, --lamb LAMB  the rate of structure loss and guilded loss
  -g GRAD_CLIP, --grad_clip GRAD_CLIP
                        clip gradients
  -K K                  K-neighborhood
  --sampling_size SAMPLING_SIZE
                        sample number
  --seed SEED           random seed
  --index_from_0 INDEX_FROM_0
                        whether the node index is from zero

Cite

If you find this code useful, please cite our paper:

@inproceedings{tu2018deep,
  title={Deep recursive network embedding with regular equivalence},
  author={Tu, Ke and Cui, Peng and Wang, Xiao and Yu, Philip S and Zhu, Wenwu},
  booktitle={Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery \& Data Mining},
  pages={2357--2366},
  year={2018},
  organization={ACM}
}

drne's People

Contributors

tadpole avatar

Watchers

James Cloos avatar Linx avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.