Skip to content

Latest commit

 

History

History
 
 

HeGAN

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 

HeGAN[KDD2019]

How to run

  • Clone the Openhgnn-DGL

    python main.py -m HeGAN -d yelp4HeGAN -t node_classification -g -1

    If you do not have gpu, set -gpu -1.

Performance: NodeClassification

dataset paper's author's ours
yelp(%) 85.24 80.01 79.35

Dataset

  • We process the yelp dataset given by yelp. It saved as dgl.heterograph and can be loaded by dgl.load_graphs

Description

  • Yelp
    Dataset Link

    Entity Statistics

    Entity #Entity
    User 1,286
    Business 2,614
    Service 2
    Star level 9
    Reservation 2
    Category 3

TrainerFlow: node_classification

model

  • HeGAN

  • 	HeGAN is a wrapper of Generator and Discriminator.
    
  • Generator

  • A Discriminator D eveluates the connectivity between the pair of nodes u and v w.r.t. a relation r.

  • Discriminator

  • A generator G samples fake node embeddings from a continuous distribution. The distribution is Gaussian distribution.

More

Contributor

Hui Han [GAMMA LAB]

If you have any questions,

Submit an issue or email to [email protected]