Back to Pytorch Geometric

Examples for Heterogeneous Data

examples/hetero/README.md

2.7.02.9 KB
Original Source

Examples for Heterogeneous Data

ExampleDescription
hetero_conv_dblp.pyShows how to use the HeteroConv(...) wrapper; Trains it for node classification on the DBLP dataset.
to_hetero_mag.pyShows how to use the to_hetero(...) functionality; Trains it for node classification on the ogb-mag dataset.
hetero_link_pred.pyShows how to use the to_hetero(...) functionality; Trains it for link prediction on the MovieLens dataset.
hgt_dblp.pyTrains a Heterogeneous Graph Transformer (HGT) model for node classification on the DBLP dataset.
hierarchical_sage.pyShows how to perform hierarchical sampling; Trains a heterogeneous GraphSAGE model for node classification on the ogb-mag dataset.
load_csv.pyShows how to create heterogeneous graphs from raw *.csv data.
metapath2vec.pyTrain an unsupervised MetaPath2Vec model; Tests embeddings for node classification on the AMiner dataset.
temporal_link_pred.pyTrains a heterogeneous GraphSAGE model for temporal link prediction on the MovieLens dataset.
bipartite_sage.pyTrains a GNN via metapaths for link prediction on the MovieLens dataset.
bipartite_sage_unsup.pyTrains a GNN via metapaths for link prediction on the large-scale TaoBao dataset.
dmgi_unsup.pyShows how to learn embeddings on the IMDB dataset using the DMGI model.
han_imdb.pyShows how to train a heterogeneous Graph Attention Network (HAN) for node classification on the IMDB dataset.
recommender_system.pyShows how to train a temporal GNN-based recommender system on the MovieLens dataset.