NodePiece

pykeen.nn.node_piece Package

NodePiece modules.

Classes

AnchorSearcher()

A method for finding the closest anchors.

ScipySparseAnchorSearcher([max_iter])

Find closest anchors using scipy.sparse.

CSGraphAnchorSearcher()

Find closest anchors using scipy.sparse.csgraph.

AnchorSelection([num_anchors])

Anchor entity selection strategy.

SingleSelection([num_anchors])

Single-step selection.

DegreeAnchorSelection([num_anchors])

Select entities according to their (undirected) degree.

MixtureAnchorSelection(selections[, ratios, ...])

A weighted mixture of different anchor selection strategies.

PageRankAnchorSelection([num_anchors, ...])

Select entities according to their page rank.

RandomAnchorSelection([num_anchors, random_seed])

Random node selection.

Tokenizer()

A base class for tokenizers for NodePiece representations.

RelationTokenizer()

Tokenize entities by representing them as a bag of relations.

AnchorTokenizer([selection, ...])

Tokenize entities by representing them as a bag of anchor entities.

PrecomputedPoolTokenizer(*[, path, url, ...])

A tokenizer using externally precomputed tokenization.

PrecomputedTokenizerLoader()

A loader for precomputed tokenization.

GalkinPrecomputedTokenizerLoader()

A loader for pickle files provided by Galkin et al.

TokenizationRepresentation(assignment[, ...])

A module holding the result of tokenization.

NodePieceRepresentation(*, triples_factory)

Basic implementation of node piece decomposition [galkin2021].

Class Inheritance Diagram

Inheritance diagram of pykeen.nn.node_piece.anchor_search.AnchorSearcher, pykeen.nn.node_piece.anchor_search.ScipySparseAnchorSearcher, pykeen.nn.node_piece.anchor_search.CSGraphAnchorSearcher, pykeen.nn.node_piece.anchor_selection.AnchorSelection, pykeen.nn.node_piece.anchor_selection.SingleSelection, pykeen.nn.node_piece.anchor_selection.DegreeAnchorSelection, pykeen.nn.node_piece.anchor_selection.MixtureAnchorSelection, pykeen.nn.node_piece.anchor_selection.PageRankAnchorSelection, pykeen.nn.node_piece.anchor_selection.RandomAnchorSelection, pykeen.nn.node_piece.tokenization.Tokenizer, pykeen.nn.node_piece.tokenization.RelationTokenizer, pykeen.nn.node_piece.tokenization.AnchorTokenizer, pykeen.nn.node_piece.tokenization.PrecomputedPoolTokenizer, pykeen.nn.node_piece.loader.PrecomputedTokenizerLoader, pykeen.nn.node_piece.loader.GalkinPrecomputedTokenizerLoader, pykeen.nn.node_piece.representations.TokenizationRepresentation, pykeen.nn.node_piece.representations.NodePieceRepresentation