matchzoo.models.arcii

An implementation of ArcII Model.

Module Contents

class matchzoo.models.arcii.ArcII

Bases: matchzoo.engine.base_model.BaseModel

ArcII Model.

Examples: >>> model = ArcII() >>> model.params[‘embedding_output_dim’] = 300 >>> model.params[‘kernel_1d_count’] = 32 >>> model.params[‘kernel_1d_size’] = 3 >>> model.params[‘kernel_2d_count’] = [16, 32] >>> model.params[‘kernel_2d_size’] = [[3, 3], [3, 3]] >>> model.params[‘pool_2d_size’] = [[2, 2], [2, 2]] >>> 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)

Build model structure.

ArcII has the desirable property of letting two sentences meet before their own high-level representations mature.

forward(self, inputs)

Forward.

classmethod _make_conv_pool_block(cls, in_channels:int, out_channels:int, kernel_size:tuple, activation:nn.Module, pool_size:tuple)

Make conv pool block.