search

datawhalechina / torch-rechub

A Lighting Pytorch Framework for Recommendation Models, Easy-to-use and Easy-to-extend.
MIT License
374 stars 69 forks source link

get_auto_embedding_dim代码错误 #69

Open yqstar opened 1 month ago

yqstar commented 1 month ago

https://github.com/datawhalechina/torch-rechub/blob/main/torch_rechub/utils/data.py中get_auto_embedding_dim的np.floor(6 np.pow(num_classes, 0.26))应该改成np.floor(6 np.pow(num_classes, 0.25))吧。