DistMultLiteral

class DistMultLiteral(triples_factory: TriplesNumericLiteralsFactory, embedding_dim: int = 50, input_dropout: float = 0.0, **kwargs)[来源]

基础类：LiteralModel

一个实现了LiteralE模型的实现，使用了来自[kristiadi2018]的DistMult交互。

初始化模型。

Parameters:

  • triples_factory (TriplesNumericLiteralsFactory) – （训练）三元组工厂

  • embedding_dim (int) – 嵌入维度

  • input_dropout (float) – 输入dropout，参见 DistMultCombination.__init__()

  • kwargs – 传递给 LiteralModel.__init__() 的额外基于关键字的参数

属性摘要

hpo_default

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

loss_default_kwargs

默认损失函数类的默认参数

属性文档

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

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

loss_default_kwargs: ClassVar[Mapping[str, Any]] = {'margin': 0.0}

默认损失函数类的默认参数