SingleSelection
- class SingleSelection(num_anchors=32)[source]
Bases:
pykeen.nn.node_piece.anchor_selection.AnchorSelection
,abc.ABC
Single-step selection.
Initialize the strategy.
- Parameters
num_anchors (
int
) – the number of anchor nodes to select. # TODO: allow relative
Methods Summary
__call__
(edge_index[, known_anchors])Select anchor nodes.
rank
(edge_index)Rank nodes.
Methods Documentation