search

librahu / HERec

Source code for TKDE 2018 "Heterogeneous information network embedding for recommendation"
166 stars 61 forks source link

Question about using other datasets #13

Open lux0818 opened 5 years ago

lux0818 commented 5 years ago

Hello professor,thanks for your code. I have a question is that ,when I use DB movie datasets, I change the metapaths ,but when I run HERec ,it has keyError, so, how to config the 'ratedim','itemdim' and 'userdim'? should I change it when I use a bigger dataset?

librahu commented 5 years ago

Of course. 'ratedim','itemdim' and 'userdim' are the hyper-parameters.

lux0818 commented 5 years ago

Thanks for your reply! can you tell me how to change it when I use Douban movies ande Douban music datasets?

librahu commented 5 years ago

They are tuned via the performance.