Improving-RNN-recommendation-model icon indicating copy to clipboard operation
Improving-RNN-recommendation-model copied to clipboard
verified publisher icon kwonmha

May I know the command line parameter when running LSTM_weight_tying(dim:100)

Open shwangdev opened this issue 6 years ago • 5 comments

May I know the command line parameter when running LSTM_weight_tying(dim:100)

shwangdev avatar Apr 09 '19 12:04 shwangdev

Just add --tying and set --r_emb, --r_l with same size. Sorry for bad README lol.

kwonmha avatar Apr 09 '19 13:04 kwonmha

And what is your value for --r_emb and --r_l? dim 100?

shwangdev avatar Apr 09 '19 13:04 shwangdev

Depends on the dataset but 100 worked best on MovieLens 1M.

kwonmha avatar Apr 09 '19 14:04 kwonmha

You did not change --max_length value?

shwangdev avatar Apr 09 '19 14:04 shwangdev

I didn't

kwonmha avatar Apr 09 '19 15:04 kwonmha