WeisfeilerLehmanInitializer

class WeisfeilerLehmanInitializer(*, color_initializer=None, color_initializer_kwargs=None, shape=32, edge_index=None, num_entities=None, mapped_triples=None, triples_factory=None, **kwargs)

Bases: PretrainedInitializer

An initializer based on an encoding of categorical colors from the Weisfeiler-Lehman algorithm.

Initialize the initializer.

Parameters:

• color_initializer (Union[str, Callable[[FloatTensor], FloatTensor], None]) – the initializer for initialization color representations, or a hint thereof

• color_initializer_kwargs (Optional[Mapping[str, Any]]) – additional keyword-based parameters for the color initializer

• shape (Union[int, Sequence[int]]) – the shape to use for the color representations

• edge_index (Optional[LongTensor]) – shape: (2, m) the edge index

• num_entities (Optional[int]) – the number of entities. can be inferred

• mapped_triples (Optional[LongTensor]) – shape: (m, 3) the Id-based triples

• triples_factory (Optional[CoreTriplesFactory]) – the triples factory

• kwargs – additional keyword-based parameters passed to pykeen.utils.iter_weisfeiler_lehman()