openhgnn.trainerflow.HeGANTrainer

class HeGANTrainer(args)

Node classification flows. Supported Model: HeGAN Supported Dataset：yelp The task is to classify the nodes of HIN(Heterogeneous Information Network). Note: If the output dim is not equal the number of classes, a MLP will follow the gnn model.