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

Make it compatible with tensorflow-2.5.0 #19

Closed colinyoyo26 closed 3 years ago

colinyoyo26 commented 3 years ago

Bump numpy from 1.16.2 to 1.19.2, pandas from 0.23.4 to 1.3.1 and scikit-learn from 0.20.0 to 0.24.2 also.