AutoSF

class AutoSF(embedding_dim: int = 256, num_components: int = 4, coefficients: Sequence[tuple[int, int, int, Literal[-1, 1]]] = [(0, 0, 0, 1), (1, 1, 1, 1), (2, 2, 2, 1), (3, 3, 3, 1), (1, 2, 3, -1), (3, 1, 1, -1)], embedding_kwargs: Mapping[str, Any] | None = None, **kwargs)

基础类：ERModel

来自[zhang2020]的AutoSF实现。

AutoSF模型结合了一个或多个pykeen.nn.Embedding`s 用于实体和关系 与 一个 :class:`pykeen.nn.AutoSFInteraction 描述其交互。

通过pykeen.nn.AutoSFInteraction交互初始化AutoSF。

注意

此变体使用num_components实体和关系表示，并共享配置。 系数应仅在\([0, num_components)\)范围内。

Parameters:

• embedding_dim (int) – 每个块的实体嵌入维度 \(d\)。

• num_components (int) – 组件/块的数量。

• coefficients (Sequence[tuple[int, int, int, Literal[-1, 1]]]) – 决定块结构的系数，参见 pykeen.nn.AutoSFInteraction。

• embedding_kwargs (Mapping[str, Any] | None) – 传递给实体和关系表示的关键字参数

• kwargs – 传递给pykeen.models.ERModel的剩余关键字参数。

属性摘要

hpo_default

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

属性文档

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

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