NTMC-Community / MatchZoo

Facilitating the design, comparison and sharing of deep text matching models.
Apache License 2.0
3.82k stars 898 forks source link

after i trained the matchlstm model, how can i Special the input layer and out layer for the predict function #812

Closed datalee closed 4 years ago

datalee commented 4 years ago

Describe the Question

after i trained the matchlstm model, how can i Special the input layer and out layer for the predict function

Describe your attempts

like this

model = mz.load_model('./xxx.h5')

embedding_layer_model = Model(inputs=model.backend.get_layer('text_left').input, outputs=model.backend.get_layer('lstm_left').output)

datalee commented 4 years ago

i have solve it,thks