openhgnn.trainerflow.NodeClassification

class NodeClassification(args)[源代码]

Node classification flow, The task is to classify the nodes of target nodes. Note: If the output dim is not equal the number of classes, we will modify the output dim with the number of classes.