drawbridge / keras-mmoe

A TensorFlow Keras implementation of "Modeling Task Relationships in Multi-task Learning with Multi-gate Mixture-of-Experts" (KDD 2018)
MIT License
681 stars 217 forks source link

Requirements Update #2

Closed alvin319 closed 5 years ago

alvin319 commented 5 years ago

@yogixRush mentioned a Keras package issue in #1. So I've updated the requirements.txt and tested both demo files. They both seem to work fine.