zyx423/Graph-Embeddding

GAE.py中进行模型的训练和测试的时候为什么没有设置model.train(), model.eval()?

Arya-bioinformatics opened this issue · 4 comments

GAE.py中进行模型的训练和测试的时候为什么没有设置model.train(), model.eval()?

感谢及时回复~
link prediction,用已知的adj和embedding的内积预测得到的adj做bce loss,应该算是有监督的吧?

感谢指正~
刚又仔细看了一下GAE的原始论文,确实是无监督。类似于矩阵分解,用adj_pred去逼近原始的adj,目的是获取node embedding,进行下游任务。