Basic-Multi-task-Learning

This is a repository for Multi-task learning with toy data in Pytorch and Tensorflow

Input data: Two synthetic regression tasks generated according to: Ma et al. (KDD 2018) "Modeling Task Relationships in Multi-task Learning with Multi-gate Mixture-of-Experts"

Network: One shared layer Two task specific towers

Optimizer: Adam

Regularization: Dropout