Source code for pygod.utils.score_converter
# -*- coding: utf-8 -*-
"""Outlier Score Converters
"""
# Author: Kay Liu <zliu234@uic.edu>
# License: BSD 2 clause
[docs]
def to_edge_score(score, edge_index):
"""Convert outlier node score to outlier edge score by averaging the
scores of two nodes connected by an edge.
Parameters
----------
score : torch.Tensor
The node score.
edge_index : torch.Tensor
The edge index.
Returns
-------
score : torch.Tensor
The edge score.
"""
score = (score[edge_index[0]] + score[edge_index[1]]) / 2
return score
[docs]
def to_graph_score(score):
"""Convert outlier node score to outlier graph score by averaging
the scores of all nodes in a graph.
Parameters
----------
score : torch.Tensor
The node score.
Returns
-------
score : torch.Tensor
The graph score.
"""
return score.mean(dim=-1)