How can I use the NRMS model ?

RUCAIBox / RecBole2.0

An up-to-date, comprehensive and flexible recommendation library

MIT License

174 stars 22 forks source link

How can I use the NRMS model ? #8

Open shamanez opened 1 year ago

Sherry-XLL commented 1 year ago

Hello @shamanez, thanks for your attention to RecBole 2.0!

NRMS is one of the models implemented in RecBole-TRM, and you can find the implementation in nrms.py. If you want to use the NRMS model, you can just follow the instructions in that repository as follows:

python run_recbole_trm.py -m [model] -d [dataset]

Also, if you have further questions, creating an issue in RecBole-TRM is recommended.

shamanez commented 1 year ago

HI,

But I found this issue. It says for the mind dataset it doesn't work. can you please help me with this ?

https://github.com/RUCAIBox/RecBole-TRM/issues/2