search

dongyuanjushi / LightLM

22 stars 2 forks source link

Hyper-parameters on Beauty and Yelp #3

Open woriazzc opened 8 months ago

woriazzc commented 8 months ago

Hi! I'm using the same hyperparameter settings on yelp and beauty as you gave in the example on toys and I'm finding the results to be very poor. May I ask what are your hyperparameters on these two datasets?

dongyuanjushi commented 8 months ago

Hi, the hyperparamters differs only in user_cluster_num and item_cluster_num. For Yelp, those are 60 and 60 and for beauty these are 20 and 20. I will also update the training config to make it more clear for these two datasets. If you still have any problem, please feel free to reach out.