CompGCN

class CompGCN(*, triples_factory: CoreTriplesFactory, embedding_dim: int = 64, encoder_kwargs: Mapping[str, Any] | None = None, interaction: str | Interaction[Tensor, RelationRepresentation, Tensor] | None = None, interaction_kwargs: Mapping[str, Any] | None = None, **kwargs)

基础类：ERModel[Tensor, RelationRepresentation, Tensor]

CompGCN的实现来自[vashishth2020]。

该模型使用图卷积和组合函数。

初始化模型。

Parameters:

• triples_factory (CoreTriplesFactory) – 三元组工厂。

• embedding_dim (int) – 如果在encoder_kwargs中没有明确给出embedding_specification，则使用的嵌入维度。

• encoder_kwargs (Mapping[str, Any] | None) – 编码器的额外关键字参数， 参见 pykeen.nn.representation.CombinedCompGCNRepresentations。

• 交互 (Interaction) – 用作解码器的交互函数。

• interaction_kwargs (Mapping[str, Any] | None) – 用于交互函数的额外基于关键字的参数。

• kwargs – 传递给pykeen.models.ERModel的额外基于关键字的参数。

属性摘要

hpo_default

优化模型超参数的默认策略

属性文档

hpo_default: ClassVar[Mapping[str, Any]] = {'embedding_dim': {'high': 512, 'low': 32, 'q': 32, 'type': <class 'int'>}}

优化模型超参数的默认策略