search

datawhalechina / torch-rechub

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

add new ranking models: DIEN, BST, AFM #64

Closed 1985312383 closed 4 months ago

1985312383 commented 7 months ago

add a ranking model DIEN, add a ranking model BST, add a ranking model AFM

add a new parameter for ctr trainer correct and add parameter descriptions

In order to maintain consistency with the DIN model of examples to call data, the create_seq_features method has been temporarily restored in ./utils/data.py