MatchZoo icon indicating copy to clipboard operation
MatchZoo copied to clipboard

How to setting learning rate in the model params??

Open gaoliming123 opened this issue 4 years ago • 0 comments

such as model build, where we can set the learning rate

model = mz.models.MatchPyramid() model.params.update(preprocessor.context) model.params['task'] = ranking_task model.params['embedding_output_dim'] = 128 model.params['embedding_input_dim'] = preprocessor.context['embedding_input_dim'] model.params['embedding_trainable'] = True model.params['num_blocks'] = 2 model.params['kernel_count'] = [8, 16] model.params['kernel_size'] = [[5, 5], [3, 3]] model.params['dpool_size'] = [3, 3] model.params['optimizer'] = 'adam' model.params['dropout_rate'] = 0.3 model.build()
model.compile()

gaoliming123 avatar Dec 30 '20 03:12 gaoliming123