最新版的cogdl如何获取有监督节点分类的结果
huihuijiangqiang opened this issue · comments
购代产特土 commented
非常感谢你们开发并开源cogdl,请问怎样获得有监督节点分类的结果而不是只显示acc呢?
Yukuo Cen commented
感谢对cogdl的关注!请问你是想获取模型最后预测出来的结果(tensor)吗?
购代产特土 commented
嗯 是的 谢谢
tonyzhu163 commented
Hi, I have the same issue. What is the preferred way to extract the predicted result from graph node classification models?
HuangJian commented
非常感谢你们开发并开源cogdl,请问怎样获得有监督节点分类的结果而不是只显示acc呢?
请问你解决了吗
tonyzhu163 commented
Note: 直接用model 内置forward/predict进行预测
购代产特土 commented
非常感谢你们开发并开源cogdl,请问怎样获得有监督节点分类的结果而不是只显示acc呢?
请问你解决了吗
还没有解决
closeressi commented
Note: 直接用model 内置forward/predict进行预测
想问下具体怎么实现呢?
Yukuo Cen commented
可以参考这个自定义模型的例子:https://github.com/THUDM/cogdl/blob/master/examples/custom_gcn.py
把外面定义model传进experiment API。训练完成后直接用这个model对数据集做预测。