提交 f88dc512 编写于 作者: Y yelrose

Add linkprediction for ogb

上级 a9693b87
......@@ -100,9 +100,7 @@ class BaseGraph(object):
self.metrics = {
"loss": self.loss,
"top1": batch_auc_out,
"max": L.reduce_max(self.logits),
"min": L.reduce_min(self.logits)
"auc": batch_auc_out,
}
def neighbor_aggregator(self, node_repr):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册