HiveGraph

Python Framework for working with Graph Neural Networks

• Contrastive Models
  • Deep Graph Contrastive Representation Learning (GRACE)
    • GRACE
• Training Engine
  • Classification Trainer
    • ClassificationTrainer
  • Trainer
    • AutoTrainer()
  • Transductive Trainer
    • TransductiveTrainer
• I/O Utilities
  • AutoDataset
    • AutoDataset
  • Dataset Transformations
    • NormalizedDegree
• Implemented Models
  • Graph Convolutional Network (GCN)
    • GCN
  • Graph Isomorphism Network (GIN)
    • GIN
• Graph Augmentations
  • drop_feature()
• AutoModel
  • AutoModel()
• Utility Functions
  • set_seed()
  • num_graphs()