最新版的cogdl如何获取有监督节点分类的结果

[huihuijiangqiang]

非常感谢你们开发并开源cogdl，请问怎样获得有监督节点分类的结果而不是只显示acc呢？

[cenyk1230]

Hi @huihuijiangqiang,

感谢对cogdl的关注！请问你是想获取模型最后预测出来的结果（tensor）吗？

[huihuijiangqiang]

嗯 是的 谢谢

[tonyzhu163]

Hi, I have the same issue. What is the preferred way to extract the predicted result from graph node classification models?

[Hdligence]

非常感谢你们开发并开源cogdl，请问怎样获得有监督节点分类的结果而不是只显示acc呢？

请问你解决了吗

[tonyzhu163]

Note: 直接用model 内置forward/predict进行预测

[huihuijiangqiang]

非常感谢你们开发并开源cogdl，请问怎样获得有监督节点分类的结果而不是只显示acc呢？

请问你解决了吗

还没有解决

[Closeressi]

Note: 直接用model 内置forward/predict进行预测

想问下具体怎么实现呢？

[cenyk1230]

可以参考这个自定义模型的例子：https://github.com/THUDM/cogdl/blob/master/examples/custom_gcn.py
把外面定义model传进experiment API。训练完成后直接用这个model对数据集做预测。