search

meta-soul / MetaSpore

A unified end-to-end machine intelligence platform
Apache License 2.0
642 stars 118 forks source link

Merge FwFM from RecBole into MetaSpore. #199

Closed andafterall closed 2 years ago

andafterall commented 2 years ago

Merge FwFM from RecBole[https://recbole.io/docs/user_guide/model/context/fwfm.html] into MetaSpore. AUC on datasets:

  1. Criteo_d5: Train AUC: 0.7517 Test AUC: 0.7298

  2. Movielens-1m uid mid: Train AUC: 0.8911 Test AUC: 0.8660

  3. Movilelens-25m uid mid: Train AUC: 0.8918 Test AUC: 0.8376

close #192 .