ERMLPInteraction

class ERMLPInteraction(embedding_dim, hidden_dim=None)[source]

Bases: FunctionalInteraction[FloatTensor, FloatTensor, FloatTensor]

A stateful module for the ER-MLP interaction.

\[f(h, r, t) = W_2 ReLU(W_1 cat(h, r, t) + b_1) + b_2\]

Initialize the interaction module.

Parameters:
  • embedding_dim (int) – The embedding vector dimension for entities and relations.

  • hidden_dim (Optional[int]) – The hidden dimension of the MLP. Defaults to embedding_dim.

Methods Summary

func(r, t, hidden, activation, final)

Evaluate the ER-MLP interaction function.

reset_parameters()

Reset parameters the interaction function may have.

Methods Documentation

func(r, t, hidden, activation, final)

Evaluate the ER-MLP interaction function.

Parameters:
  • h (FloatTensor) – shape: (*batch_dims, dim) The head representations.

  • r (FloatTensor) – shape: (*batch_dims, dim) The relation representations.

  • t (FloatTensor) – shape: (*batch_dims, dim) The tail representations.

  • hidden (Linear) – The first linear layer.

  • activation (Module) – The activation function of the hidden layer.

  • final (Linear) – The second linear layer.

Return type:

FloatTensor

Returns:

shape: batch_dims The scores.

reset_parameters()[source]

Reset parameters the interaction function may have.