A recommendation system models based Keras
Project description
multi_task_learning
Multi-task Learning Models for Recommender Systems
This project is developed based on DeepCTR :https://github.com/shenweichen/DeepCTR.
You can easy to use the code to design your multi task learning model for multi regression or classification tasks.
Example 1
Dataset: http://archive.ics.uci.edu/ml/machine-learning-databases/adult/
Task 1: (Classification) aims to predict whether the income exceeds 50K.
Task 2: (Classification) aims to predict this person’s marital status is never married.
Example 2
Dataset: https://archive.ics.uci.edu/ml/machine-learning-databases/census-income-mld/
Preparing
| 模型 | 简介 | 论文 |
|---|---|---|
| Shared-Bottom | shared-bottom | Multitask learning(1998) |
| ESMM | Entire Space Multi-Task Model | Entire Space Multi-Task Model: An Effective Approach for Estimating Post-Click Conversion Rate(SIGIR'18) |
| MMoE | Multi-gate Mixture-of-Experts | Modeling Task Relationships in Multi-task Learning with Multi-gate Mixture-of-Experts(KDD'18) |
| CGC | Customized Gate Control | Progressive Layered Extraction (PLE): A Novel Multi-Task Learning (MTL) Model for Personalized Recommendations(RecSys '20) |
| PLE | Progressive Layered Extraction | Progressive Layered Extraction (PLE): A Novel Multi-Task Learning (MTL) Model for Personalized Recommendations(RecSys '20) |
Shared-Bottom
ESMM
MMOE
CGC
PLE
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
keras-recommenders-0.0.2.tar.gz
(10.7 kB
view hashes)
