├── .gitignore ├── LICENSE.txt ├── README.md ├── mtrec ├── __init__.py ├── functions │ ├── __init__.py │ └── feature_column.py ├── layers │ ├── __init__.py │ └── core.py └── models │ ├── __init__.py │ └── mmoe.py ├── setup.py └── test ├── data └── census │ ├── census-income.data.gz │ └── census-income.test.gz ├── mmoe_demo.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZiyaoGeng/MTRec/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZiyaoGeng/MTRec/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZiyaoGeng/MTRec/HEAD/README.md -------------------------------------------------------------------------------- /mtrec/__init__.py: -------------------------------------------------------------------------------- 1 | name = "mtrec" 2 | -------------------------------------------------------------------------------- /mtrec/functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mtrec/functions/feature_column.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZiyaoGeng/MTRec/HEAD/mtrec/functions/feature_column.py -------------------------------------------------------------------------------- /mtrec/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mtrec/layers/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZiyaoGeng/MTRec/HEAD/mtrec/layers/core.py -------------------------------------------------------------------------------- /mtrec/models/__init__.py: -------------------------------------------------------------------------------- 1 | from mtrec.models.mmoe import MMoE 2 | 3 | 4 | __all__ = ['MMoE'] -------------------------------------------------------------------------------- /mtrec/models/mmoe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZiyaoGeng/MTRec/HEAD/mtrec/models/mmoe.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZiyaoGeng/MTRec/HEAD/setup.py -------------------------------------------------------------------------------- /test/data/census/census-income.data.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZiyaoGeng/MTRec/HEAD/test/data/census/census-income.data.gz -------------------------------------------------------------------------------- /test/data/census/census-income.test.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZiyaoGeng/MTRec/HEAD/test/data/census/census-income.test.gz -------------------------------------------------------------------------------- /test/mmoe_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZiyaoGeng/MTRec/HEAD/test/mmoe_demo.py -------------------------------------------------------------------------------- /test/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZiyaoGeng/MTRec/HEAD/test/utils.py --------------------------------------------------------------------------------