DistMA
- class DistMA(embedding_dim: int = 256, entity_initializer: str | Callable[[Tensor], Tensor] | None = None, entity_initializer_kwargs: Mapping[str, Any] | None = None, entity_normalizer: str | Callable[[Tensor], Tensor] | None = None, entity_normalizer_kwargs: Mapping[str, Any] | None = None, relation_initializer: str | Callable[[Tensor], Tensor] | None = None, relation_initializer_kwargs: Mapping[str, Any] | None = None, **kwargs)[source]
基础类：
ERModel[Tensor,Tensor,Tensor]来自[shi2019]的DistMA实现。
它通过存储在
Embedding中的\(d\)维向量来建模实体和关系，并使用DistMAInteraction来获取三元组分数。初始化模型。
- Parameters:
embedding_dim (int) – 实体嵌入维度 \(d\)。
entity_initializer (str | Callable[[Tensor], Tensor] | None) – 实体初始化函数。默认为 None
entity_initializer_kwargs (Mapping[str, Any] | None) – 调用实体初始化器时要使用的关键字参数
entity_normalizer (str | Callable[[Tensor], Tensor] | None) – 实体标准化函数。默认为 None
entity_normalizer_kwargs (Mapping[str, Any] | None) – 调用实体规范化器时要使用的关键字参数
relation_initializer (str | Callable[[Tensor], Tensor] | None) – 关系初始化函数。默认为 None
relation_initializer_kwargs (Mapping[str, Any] | None) – 调用关系初始化器时要使用的关键字参数
kwargs – 传递给
pykeen.models.ERModel的剩余关键字参数。
属性摘要
优化模型超参数的默认策略
属性文档