Skip to content

Multi-Task Learning package built with tensorflow 2 (Multi-Gate Mixture of Experts, Cross-Stitch, Ucertainty Weighting)

Notifications You must be signed in to change notification settings

AmazaspShumik/mtlearn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

52 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

mtlearn

The idea behind this repository is to build a small package that contains building blocks of many modern Multi-Task Learning algorithms and then use it to reproduce papers. This is still work in progress.

Installing & Upgrading package

pip install https://github.com/AmazaspShumik/mtlearn/archive/master.zip
pip install --upgrade https://github.com/AmazaspShumik/mtlearn/archive/master.zip

Papers Reproduced:

Modeling Task Relationships in Multi-task Learning with Multi-gate Mixture-of-Experts by Jiaqi Ma et al (KDD 2018)

About

Multi-Task Learning package built with tensorflow 2 (Multi-Gate Mixture of Experts, Cross-Stitch, Ucertainty Weighting)

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages