matchzoo.models.hbmp
¶
An implementation of HBMP Model.
Module Contents¶
Classes¶
HBMP model. |
-
class
matchzoo.models.hbmp.
HBMP
(params: typing.Optional[ParamTable] = None)¶ Bases:
matchzoo.engine.base_model.BaseModel
HBMP model.
Examples
>>> model = HBMP() >>> model.params['embedding_input_dim'] = 200 >>> model.params['embedding_output_dim'] = 100 >>> model.params['mlp_num_layers'] = 1 >>> model.params['mlp_num_units'] = 10 >>> model.params['mlp_num_fan_out'] = 10 >>> model.params['mlp_activation_func'] = nn.LeakyReLU(0.1) >>> model.params['lstm_hidden_size'] = 5 >>> model.params['lstm_num'] = 3 >>> model.params['num_layers'] = 3 >>> model.params['dropout_rate'] = 0.1 >>> model.guess_and_fill_missing_params(verbose=0) >>> model.build()
-
classmethod
get_default_params
(cls) → ParamTable¶ - Returns
model default parameters.
-
build
(self)¶ Build model structure.
HBMP use Siamese arthitecture.
-
forward
(self, inputs)¶ Forward.
-
classmethod