Skip to content

[Task] Add RNN-based model apis for session-based recommendation #819

@rnyak

Description

@rnyak

Description

Currently there is no out-of-the-box api for creating an LSTM, BiLSTM and GRU architectures for a session-based (or sequential) task.

  • For LSTM , I assume we'd use tf.keras.layers.LSTM api.
  • For BiLSTM shall we use tf.keras.layers.Bidirectional?
  • For GRU, can we not port GRU4Rec model code from TF4Rec to Merlin Models?

Looks like there were two tickets before but they were closed without completion:

#732
#733

Metadata

Metadata

Type

No type

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions