https://github.com/matenure/FastGCN

The sample codes for our ICLR18 paper "FastGCN: Fast Learning with Graph Convolutional Networks via Importance Sampling""
https://github.com/matenure/FastGCN

fastgcn graph-convolutional-networks graphsage reddit

Last synced: 3 months ago
JSON representation

The sample codes for our ICLR18 paper "FastGCN: Fast Learning with Graph Convolutional Networks via Importance Sampling""

• Host: GitHub
• URL: https://github.com/matenure/FastGCN
• Owner: matenure
• Created: 2018-01-30T21:33:04.000Z (almost 7 years ago)
• Default Branch: master
• Last Pushed: 2021-03-25T03:16:19.000Z (almost 4 years ago)
• Last Synced: 2024-05-21T17:23:34.890Z (8 months ago)
• Topics: fastgcn, graph-convolutional-networks, graphsage, reddit
• Language: Python
• Size: 4.96 MB
• Stars: 515
• Watchers: 12
• Forks: 113
• Open Issues: 25
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# FastGCN

This is the Tensorflow implementation of our ICLR2018 paper: ["**FastGCN: Fast Learning with Graph Convolutional Networks via Importance Sampling**".](https://openreview.net/forum?id=rytstxWAW&noteId=ByU9EpGSf)

Instructions of the sample codes:

[For Reddit dataset]

	train_batch_multiRank_inductive_reddit_Mixlayers_sampleA.py is the final model. (precomputated the AH in the bottom layer) The original Reddit data should be transferred into the .npz format using this function: transferRedditDataFormat.

	Note: By default, this code does no sampling. To enable sampling, change `main(None)` at the bottom to `main(100)`. (The number is the sample size. You can also try other sample sizes)

	train_batch_multiRank_inductive_reddit_Mixlayers_uniform.py is the model for uniform sampling.

	train_batch_multiRank_inductive_reddit_Mixlayers_appr2layers.py is the model for 2-layer approximation.

	create_Graph_forGraphSAGE.py is used to transfer the data into the GraphSAGE format, so that users can compare our method with GraphSAGE. We also include the transferred original Cora dataset in this repository (./data/cora_graphSAGE).

[For pubmed or cora]

	train.py is the original GCN model.

 	pubmed_Mix_sampleA.py 	The dataset could be defined in the codes, for example: flags.DEFINE_string('dataset', 'pubmed', 'Dataset string.')

	pubmed_Mix_uniform.py and pubmed_inductive_appr2layers.py are similar to the ones for reddit.

	pubmed-original**.py means the codes are used for original Cora or Pubmed datasets. Users could also change their datasets by changing the data load function from load_data() to load_data_original().