search

RUCAIBox / RecBole-TRM

MIT License
13 stars 5 forks source link

Can I run this model with implicit feedback data ? #5

Open shamanez opened 1 year ago

shamanez commented 1 year ago

I am trying to run the CORE model with my custom dataset with the implicit feedback data.

The CORE documentation says I can use "BPR" loss, which supports the implicit feedback dataset.

Can I change the default loss of "CE" to the "BPR" in CORE model?

Thanks in advanced.