OpenViSTAI · Agents · Blockchain · Crypto
Submit a project
Menu
AI Models

Keras Mmoe

Keras Mmoe is a Python implementation of the multi-gate mixture-of-experts model in TensorFlow and Keras, intended for multi-task learning. It includes a census-income example and code described as extensible for further experiments.

Explore the practical guide ↓

Independent OpenViST project profile, based on linked sources.

Keras Mmoe — OpenViST project profile in AI Models.
OpenViST project illustration. See the linked sources for product details.
Understand the project

Practical overview

When it fits

Use it when you want a code-level starting point for experimenting with the MMoE approach, especially through the included census-income demo.

Who it is for

  • Developers and researchers exploring multi-task learning with TensorFlow and Keras.
  • Practitioners looking for a census-income example of a multi-gate mixture-of-experts model.

Documented uses

  • Study or run an MMoE model example on census-income data.
  • Use shared expert representations to build multi-task prediction models.
  • Explore adapting the implementation to other datasets and multi-task prediction problems.
Before you choose

Requirements & limits

The repository is public and specifies an MIT license.

  • The implementation uses Python, TensorFlow and Keras; the README lists Python 3.6 and dependencies including TensorFlow and Scikit-learn.
  • The included example uses the UCI census-income dataset.
  • The repository cautions that its results do not exactly reproduce those in the paper.

Check current requirements in the official documentation. Source-based guidance does not mean OpenViST has run or independently tested this project.

Source-based practical guide

Set up and run the census-income demo

Install the repository's listed dependencies and start its included example.

Before you start

  • Python 3.6, as listed in the getting-started requirements.
  • TensorFlow and Scikit-learn, listed among the dependencies in requirements.txt.

Steps

  1. Clone the repository.

  2. Install the listed dependencies with pip install -r requirements.txt.

  3. Run python census_income_demo.py.

Expected outcome

Starts the included model example using the census-income dataset.

Guide limitations

  • The repository cautions that it does not reproduce the paper's exact results.
  • Python 2.7 compatibility is also stated, but the getting-started requirements specify Python 3.6.
Documentation 1 ↗
Evidence and transparency

Sources & review

Official repository ↗
Source record details
Source availability
Active
Sources checked
7 November 2026
Original website title
drawbridge/keras-mmoe · Official homepage title · retained in its source language
How we review sourcesSuggest a correction