matchzoo.models.bimpm

An implementation of BiMPM Model.

Module Contents

class matchzoo.models.bimpm.BiMPM

Bases: matchzoo.engine.base_model.BaseModel

BiMPM Model.

Reference: - https://github.com/galsang/BIMPM-pytorch/blob/master/model/BIMPM.py

Examples

>>> model = BiMPM()
>>> model.params['num_perspective'] = 4
>>> model.guess_and_fill_missing_params(verbose=0)
>>> model.build()
classmethod get_default_params(cls)
Returns:model default parameters.
classmethod get_default_padding_callback(cls)
Returns:Default padding callback.
build(self)

Make function layers.

forward(self, inputs)

Forward.

reset_parameters(self)

Init Parameters.

dropout(self, v)

Dropout Layer.

matchzoo.models.bimpm.mp_matching_func(v1, v2, w)

Basic mp_matching_func.

Parameters:
  • v1 – (batch, seq_len, hidden_size)
  • v2 – (batch, seq_len, hidden_size) or (batch, hidden_size)
  • w – (num_psp, hidden_size)
Returns:

(batch, num_psp)
matchzoo.models.bimpm.mp_matching_func_pairwise(v1, v2, w)

Basic mp_matching_func_pairwise.

Parameters:
  • v1 – (batch, seq_len1, hidden_size)
  • v2 – (batch, seq_len2, hidden_size)
  • w – (num_psp, hidden_size)

:param num_psp :return: (batch, num_psp, seq_len1, seq_len2)

matchzoo.models.bimpm.attention(v1, v2)

Attention.

Parameters:
  • v1 – (batch, seq_len1, hidden_size)
  • v2 – (batch, seq_len2, hidden_size)
Returns:

(batch, seq_len1, seq_len2)
matchzoo.models.bimpm.div_with_small_value(n, d, eps=1e-08)

Small values are replaced by 1e-8 to prevent it from exploding.

Parameters:
  • n – tensor
  • d – tensor
Returns:

n/d: tensor