← all repositories

drawbridge/keras-mmoe

A TensorFlow Keras implementation of the Multi-gate Mixture-of-Experts model for multi-task learning from KDD 2018.

733 stars Python ML Frameworks
keras-mmoe
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

This repository provides a Keras-based implementation of the Multi-gate Mixture-of-Experts (MMoE) architecture, a deep learning model that learns relationships between multiple tasks by leveraging shared expert networks with task-specific gating mechanisms. The implementation includes example demonstrations using the UCI census-income dataset and is designed to be extensible for other multi-task learning scenarios.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.