DistMultCombination

class DistMultCombination(entity_embedding_dim, literal_embedding_dim, input_dropout=0.0)[source]

Bases: pykeen.nn.combinations.ParameterizedRealCombination

The linear/dropout combination used in pykeen.models.DistMultLiteral.

Instantiate the ParameterizedRealCombination with a LinearDropout.

Parameters

  • entity_embedding_dim (int) – The dimension of the entity representations to which literals are concatenated

  • literal_embedding_dim (int) – The dimension of the literals that are concatenated

  • input_dropout (float) – The dropout probability of an element to be zeroed.

This class does not use an activation in the LinearDropout as described by [kristiadi2018].