tch-geometric

Pytorch Geometric extension library

Features • Installation • Examples

Features

Pytorch Geometric extension library with additional graph sampling algorithms.

Supports:

Node2Vec (random_walk)
Temporal Random Walk (temporal_random_walk)
Biased Temporal Random Walk (CTDNE) (biased_tempo_random_walk)
Negative Sampling (negative_sample_neighbors_homogenous and negative_sample_neighbors_heterogenous)
GraphSAGE budget sampling (budget_sampling)
Temporal Heterogenous Graph Transformer (HGT) sampling (hgt_sampling)
GraphSAGE (neighbor_sampling_heterogenous and neighbor_sampling_homogenous)

Check examples folder