PageRankAnchorSelection

class PageRankAnchorSelection(num_anchors: int = 32, **kwargs)[source]

Bases: SingleSelection

Select entities according to their page rank.

See also

http://web.stanford.edu/class/cs224w/slides/04-pagerank.pdf

Initialize the selection strategy.

Parameters:

  • num_anchors (int) – the number of anchors to select

  • kwargs – additional keyword-based parameters passed to page_rank().

Methods Summary

iter_extra_repr()

Extra components for __repr__.

rank(edge_index)

Rank nodes.

Methods Documentation

iter_extra_repr() → Iterable[str][source]

Extra components for __repr__.

Return type:

Iterable[str]

rank(edge_index: ndarray) → ndarray[source]

Rank nodes.

Parameters:

edge_index (ndarray) – shape: (m, 2) the edge_index, i.e., adjacency list.

Returns:

(n,) the node IDs sorted decreasingly by anchor selection preference.

Return type:

ndarray