BUPT-GAMMA / OpenHGNN

This is an open-source toolkit for Heterogeneous Graph Neural Network(OpenHGNN) based on DGL.
Apache License 2.0
816 stars 139 forks source link

add RoHe #187

Closed CoincidenceBoy closed 12 months ago

CoincidenceBoy commented 1 year ago

Description

Checklist

Please feel free to remove inapplicable items for your PR.

Changes

  1. config.py & config.ini
  2. experiment.py
    • Experiment.field: specific_trainerflow
  3. trainerflow/

    1. __init.py

      • SUPPORTED_FLOWS
      • from .RoHe_trainer import RoHeTrainer
    2. RoHe_trainer.py

      • RoHeTrainer
  4. modules/
    1. __init.py
      • SUPPORTED_MODELS
      • from .RoHe import RoHe
    2. RoHe.py
      • RoHe