怎么获取模型的acc、recall、F1或者分类报告？ about cogdl HOT 4 CLOSED

非常感谢关注CogDL！请问这边是只需要有监督节点分类的结果 还是 有监督/无监督 都需要呢？无监督这边的话CogDL里面目前已经保存了生成的embedding

from cogdl.

非常感谢回复，您好，我这边，有监督/无监督都需要。

from cogdl.

你好，对于有监督的情况，可以先训练保存模型，再得到结果，比如 gcn + cora:

1. python scripts/train.py --task node_classification --model gcn --dataset cora --save_model <model_path>
2. python scripts/train.py --task node_classification --model gcn --dataset cora --checkpoint <model_path> --inference
   则模型的预测结果会被保存在 gcn_cora.pred, 可以通过 torch.load读取

对于无监督的情况，每次训练得到的embedding保存在./embedding下，可以通过np.load读取，然后您可以训练一个 linear classifier 然后根据个人需要得到分类结果

from cogdl.

非常感谢

from cogdl.