search

hidasib / GRU4Rec

GRU4Rec is the original Theano implementation of the algorithm in "Session-based Recommendations with Recurrent Neural Networks" paper, published at ICLR 2016 and its follow-up "Recurrent Neural Networks with Top-k Gains for Session-based Recommendations". The code is optimized for execution on the GPU.
Other
754 stars 223 forks source link

Saving a model #4

Closed pabloloyola closed 7 years ago

pabloloyola commented 7 years ago

Hi, if I want save a model, should I care about the following objects ? self.Wx, self.Wh, self.Wrz, self.Bh, self.H

thank you!

hidasib commented 7 years ago

Use pickle to save the whole GRU object.

loretoparisi commented 7 years ago

@pabloloyola @hidasib is it possibile to save using the dump like theano.misc.pkl_utils.dump(gru,...), supposed that the whole gru object is serializable?