elephas icon indicating copy to clipboard operation
elephas copied to clipboard

Published 20 hours ago • verified publisher icon maxpumperla

Does support LSTM or GRU models?

Open ghorbani-mohammad opened this issue 3 years ago • 5 comments

Hi,

Does this package support LSTM or GRU models for training on spark?

Thanks for your great work :)

ghorbani-mohammad avatar Apr 08 '21 19:04 ghorbani-mohammad

Thank you! Not exactly - there are several other open issues (https://github.com/maxpumperla/elephas/issues/141, https://github.com/maxpumperla/elephas/issues/85) regarding LSTM support. It's something I could add but I would definitely need some help.

danielenricocahall avatar Apr 23 '21 12:04 danielenricocahall

Hi, was LSTM and Bidirectional LSTM added in the recent release?

paravatha avatar Aug 14 '21 15:08 paravatha

Hi @paravatha it was not - that's still an ongoing effort. Feel free to start it and we can work on it together if you have any ideas!

danielenricocahall avatar Aug 17 '21 01:08 danielenricocahall

Hello! Was LSTM added in the meantime? I tried to use a LSTM layer and fit it with a rdd and it did run without any error. Do you think that it was a case of luck and the results might be correct?

rebelobru avatar Mar 18 '22 15:03 rebelobru

Hello @rebelobru! LSTM wasn't added - it may work in the sense that a model will train, but I haven't thought through if it would actually work in the current paradigm. Out of curiosity, how did the results look in the LSTM model trained with elephas?

danielenricocahall avatar Sep 07 '22 14:09 danielenricocahall

Moved this issue to the new fork: https://github.com/danielenricocahall/elephas/issues/10. Closing this for now but still on the radar!

danielenricocahall avatar Oct 11 '22 13:10 danielenricocahall