THUDM / CogDL

CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)

Home Page:https://cogdl.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

huihuijiangqiang opened this issue · comments

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

Hi @huihuijiangqiang,

感谢对cogdl的关注!请问你是想获取模型最后预测出来的结果(tensor)吗?

嗯 是的 谢谢

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

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

请问你解决了吗

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

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

请问你解决了吗

还没有解决

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

想问下具体怎么实现呢?

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