hogunpark / TSGNET

TSGNet (Temporal-Static-Graph-Net)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

TSGNet

This is a Pytorch implementation of TSGNet (Temporal-Static-Graph-Net), as described in our paper:

Hogun Park, Jennifer Neville, Exploiting Interaction Links for Node Classification with Deep Graph Neural Networks (IJCAI 2019)

Usage

Example Usage $main.py --dataname "imdb"

Full Command List The full list of command-line options is available with $main.py --help

Requirements

We tested the codes in the following environment, but it may work in their previous versions.

  • Python 3.6
  • Pytorch 1.7.0
  • Python Geometric 1.6.3

Cite

If you find TSGNet useful in your research, we ask that you cite the following paper::

@inproceedings{parkneville2019,
  title     = {Exploiting Interaction Links for Node Classification with Deep Graph Neural Networks},
  author    = {Park, Hogun and Neville, Jennifer},
  booktitle = {Proceedings of the Twenty-Eighth International Joint Conference on
               Artificial Intelligence, {IJCAI-19}},
  publisher = {International Joint Conferences on Artificial Intelligence Organization},
  pages     = {3223--3230},
  year      = {2019},
  month     = {7},
  doi       = {10.24963/ijcai.2019/447},
  url       = {https://doi.org/10.24963/ijcai.2019/447},
}

Misc

Datasets and Implementation of layers are following interfaces of Pytorch geometric.

About

TSGNet (Temporal-Static-Graph-Net)

License:MIT License


Languages

Language:Python 100.0%