WeisfeilerLehmanInitializer

class WeisfeilerLehmanInitializer(*, color_initializer: str | Callable[[Tensor], Tensor] | None = None, color_initializer_kwargs: Mapping[str, Any] | None = None, shape: int | Sequence[int] = 32, edge_index: Tensor | None = None, num_entities: int | None = None, mapped_triples: Tensor | None = None, triples_factory: CoreTriplesFactory | None = None, **kwargs)[source]

基础类：PretrainedInitializer

基于Weisfeiler-Lehman算法编码的分类颜色的初始化器。

初始化初始化器。

Parameters:

  • color_initializer (str | Callable[[Tensor], Tensor] | None) – 用于初始化颜色表示的初始化器，或其提示

  • color_initializer_kwargs (Mapping[str, Any] | None) – 颜色初始化器的额外基于关键字的参数

  • shape (int | Sequence[int]) – 用于颜色表示的形状

  • edge_index (Tensor | None) – 形状: (2, m) 边索引

  • num_entities (int | None) – 实体的数量。可以推断

  • mapped_triples (Tensor | None) – 形状: (m, 3) 基于ID的三元组

  • triples_factory (CoreTriplesFactory | None) – 三元组工厂

  • kwargs – 传递给 pykeen.utils.iter_weisfeiler_lehman() 的额外基于关键字的参数