Retrieval-based_Chatbot icon indicating copy to clipboard operation
Retrieval-based_Chatbot copied to clipboard

Possible to Implement bi-LSTM?

Open thombrey-vinil opened this issue 7 years ago • 1 comments

Is it possible to implement a bidirectional LSTM just by changing parameter 'bidirectional = True' in the code?

Please let me know if this works.

thombrey-vinil avatar Aug 22 '18 12:08 thombrey-vinil

I think it should work in general. But you might need to adjust the way you access and slice the dimensions, because the output of a bidirectional LSTM will have a different dimension than the output of a unidirectional LSTM, I suppose. See the PyTorch documentation for details. But I never tried it... Also, using a bidirectional LSTM might lead to even more overfitting.

Janinanu avatar Nov 06 '18 13:11 Janinanu