MutiTaskLearning

PLE RecSys2020最佳长论文奖 Progressive Layered Extraction (PLE): A Novel Multi-Task Learning (MTL) Model for Personalized Recommendations 的非官方实现版本。

参考了这个代码实现的:

https://github.com/drawbridge/keras-mmoe

由于目前还没有在多目标优化的数据集上大规模测试,暂时只是一个代码实现。没有复现出原论文的效果。

如何开始?

1、pip insall -r requirements.txt 说明:我这个代码是兼容tensorflow1.x的。我本身的版本是tensorflow 2.x

2、 python train_ple.py

just run !