datawhalechina / torch-rechub

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

multi-task mmoe ali_ccp 全量数据集,ctr任务测试auc=0.5 #50

Open scriptboy1990 opened 1 year ago

scriptboy1990 commented 1 year ago

下载了全量数据集,参数的话,只改了两个tower的dims为[256,128,64],其他没动,不管设置什么seed,最后出来的ctr auc总是0.5啊,cvr的看着还算正常,接近0.6.

Hjh233 commented 8 months ago

我没有改超参数,训练一个epoch之后,cvr的auc=1.0,ctr的auc是0.62

KrisZQ commented 6 months ago

模型训练完毕后,使用.predict对训练集进行预测后,使用预测结果计算auc的值与使用.evaluate评估的auc值不一致!不晓得问题出在哪里。