├── .github └── workflows │ ├── codeql-analysis.yml │ └── continuous_integration.yml ├── .gitignore ├── .travis.yml ├── LICENSE ├── README.md ├── deep_recommenders ├── __init__.py ├── datasets │ ├── __init__.py │ ├── cora.py │ ├── movielens.py │ └── synthetic_for_multi_task.py ├── estimator │ ├── __init__.py │ └── models │ │ ├── __init__.py │ │ ├── feature_interaction │ │ ├── __init__.py │ │ ├── dnn.py │ │ └── fm.py │ │ ├── multi_task_learning │ │ ├── __init__.py │ │ ├── esmm.py │ │ └── mixture_of_experts.py │ │ └── ranking │ │ ├── __init__.py │ │ ├── deepfm.py │ │ ├── fnn.py │ │ └── wide_and_deep.py └── keras │ ├── __init__.py │ └── models │ ├── __init__.py │ ├── nlp │ ├── __init__.py │ ├── multi_head_attention.py │ └── transformer.py │ ├── ranking │ ├── __init__.py │ ├── dcn.py │ ├── deepfm.py │ ├── din.py │ ├── fm.py │ └── xdeepfm.py │ └── retrieval │ ├── __init__.py │ ├── factorized_top_k.py │ ├── gcn.py │ └── sbcnm.py ├── examples ├── train_deepfm_on_movielens_estimator.py ├── train_deepfm_on_movielens_keras.py ├── train_fm_on_movielens_estimator.py ├── train_fnn_on_movielens_estimator.py ├── train_gcn_on_cora_keras.py ├── train_mmoe_on_synthetic_estimator.py ├── train_transformer_on_imdb_keras.py └── train_wdl_on_movielens_estimator.py ├── requirements.txt └── tests ├── datasets ├── test_movielens.py └── test_synthetic_for_multi_task.py ├── estimator ├── test_esmm.py ├── test_fm.py └── test_mixture_of_experts.py ├── keras ├── test_dcn.py ├── test_deepfm.py ├── test_din.py ├── test_factorized_top_k.py ├── test_fm.py ├── test_gcn.py ├── test_sbcnm.py ├── test_transformer.py └── test_xdeepfm.py └── testing.sh /.github/workflows/codeql-analysis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/.github/workflows/codeql-analysis.yml -------------------------------------------------------------------------------- /.github/workflows/continuous_integration.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/.github/workflows/continuous_integration.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/.gitignore -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/.travis.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/README.md -------------------------------------------------------------------------------- /deep_recommenders/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/python3 2 | # -*- coding: utf-8 -*- 3 | -------------------------------------------------------------------------------- /deep_recommenders/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/datasets/__init__.py -------------------------------------------------------------------------------- /deep_recommenders/datasets/cora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/datasets/cora.py -------------------------------------------------------------------------------- /deep_recommenders/datasets/movielens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/datasets/movielens.py -------------------------------------------------------------------------------- /deep_recommenders/datasets/synthetic_for_multi_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/datasets/synthetic_for_multi_task.py -------------------------------------------------------------------------------- /deep_recommenders/estimator/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/python3 2 | # -*- coding: utf-8 -*- 3 | -------------------------------------------------------------------------------- /deep_recommenders/estimator/models/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/python3 2 | # -*- coding: utf-8 -*- 3 | -------------------------------------------------------------------------------- /deep_recommenders/estimator/models/feature_interaction/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/estimator/models/feature_interaction/__init__.py -------------------------------------------------------------------------------- /deep_recommenders/estimator/models/feature_interaction/dnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/estimator/models/feature_interaction/dnn.py -------------------------------------------------------------------------------- /deep_recommenders/estimator/models/feature_interaction/fm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/estimator/models/feature_interaction/fm.py -------------------------------------------------------------------------------- /deep_recommenders/estimator/models/multi_task_learning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/estimator/models/multi_task_learning/__init__.py -------------------------------------------------------------------------------- /deep_recommenders/estimator/models/multi_task_learning/esmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/estimator/models/multi_task_learning/esmm.py -------------------------------------------------------------------------------- /deep_recommenders/estimator/models/multi_task_learning/mixture_of_experts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/estimator/models/multi_task_learning/mixture_of_experts.py -------------------------------------------------------------------------------- /deep_recommenders/estimator/models/ranking/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/estimator/models/ranking/__init__.py -------------------------------------------------------------------------------- /deep_recommenders/estimator/models/ranking/deepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/estimator/models/ranking/deepfm.py -------------------------------------------------------------------------------- /deep_recommenders/estimator/models/ranking/fnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/estimator/models/ranking/fnn.py -------------------------------------------------------------------------------- /deep_recommenders/estimator/models/ranking/wide_and_deep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/estimator/models/ranking/wide_and_deep.py -------------------------------------------------------------------------------- /deep_recommenders/keras/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/python3 2 | # -*- coding: utf-8 -*- 3 | -------------------------------------------------------------------------------- /deep_recommenders/keras/models/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/python3 2 | # -*- coding: utf-8 -*- 3 | -------------------------------------------------------------------------------- /deep_recommenders/keras/models/nlp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/nlp/__init__.py -------------------------------------------------------------------------------- /deep_recommenders/keras/models/nlp/multi_head_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/nlp/multi_head_attention.py -------------------------------------------------------------------------------- /deep_recommenders/keras/models/nlp/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/nlp/transformer.py -------------------------------------------------------------------------------- /deep_recommenders/keras/models/ranking/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/ranking/__init__.py -------------------------------------------------------------------------------- /deep_recommenders/keras/models/ranking/dcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/ranking/dcn.py -------------------------------------------------------------------------------- /deep_recommenders/keras/models/ranking/deepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/ranking/deepfm.py -------------------------------------------------------------------------------- /deep_recommenders/keras/models/ranking/din.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/ranking/din.py -------------------------------------------------------------------------------- /deep_recommenders/keras/models/ranking/fm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/ranking/fm.py -------------------------------------------------------------------------------- /deep_recommenders/keras/models/ranking/xdeepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/ranking/xdeepfm.py -------------------------------------------------------------------------------- /deep_recommenders/keras/models/retrieval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/retrieval/__init__.py -------------------------------------------------------------------------------- /deep_recommenders/keras/models/retrieval/factorized_top_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/retrieval/factorized_top_k.py -------------------------------------------------------------------------------- /deep_recommenders/keras/models/retrieval/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/retrieval/gcn.py -------------------------------------------------------------------------------- /deep_recommenders/keras/models/retrieval/sbcnm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/deep_recommenders/keras/models/retrieval/sbcnm.py -------------------------------------------------------------------------------- /examples/train_deepfm_on_movielens_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/examples/train_deepfm_on_movielens_estimator.py -------------------------------------------------------------------------------- /examples/train_deepfm_on_movielens_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/examples/train_deepfm_on_movielens_keras.py -------------------------------------------------------------------------------- /examples/train_fm_on_movielens_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/examples/train_fm_on_movielens_estimator.py -------------------------------------------------------------------------------- /examples/train_fnn_on_movielens_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/examples/train_fnn_on_movielens_estimator.py -------------------------------------------------------------------------------- /examples/train_gcn_on_cora_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/examples/train_gcn_on_cora_keras.py -------------------------------------------------------------------------------- /examples/train_mmoe_on_synthetic_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/examples/train_mmoe_on_synthetic_estimator.py -------------------------------------------------------------------------------- /examples/train_transformer_on_imdb_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/examples/train_transformer_on_imdb_keras.py -------------------------------------------------------------------------------- /examples/train_wdl_on_movielens_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/examples/train_wdl_on_movielens_estimator.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/requirements.txt -------------------------------------------------------------------------------- /tests/datasets/test_movielens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/datasets/test_movielens.py -------------------------------------------------------------------------------- /tests/datasets/test_synthetic_for_multi_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/datasets/test_synthetic_for_multi_task.py -------------------------------------------------------------------------------- /tests/estimator/test_esmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/estimator/test_esmm.py -------------------------------------------------------------------------------- /tests/estimator/test_fm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/estimator/test_fm.py -------------------------------------------------------------------------------- /tests/estimator/test_mixture_of_experts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/estimator/test_mixture_of_experts.py -------------------------------------------------------------------------------- /tests/keras/test_dcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/keras/test_dcn.py -------------------------------------------------------------------------------- /tests/keras/test_deepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/keras/test_deepfm.py -------------------------------------------------------------------------------- /tests/keras/test_din.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/keras/test_din.py -------------------------------------------------------------------------------- /tests/keras/test_factorized_top_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/keras/test_factorized_top_k.py -------------------------------------------------------------------------------- /tests/keras/test_fm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/keras/test_fm.py -------------------------------------------------------------------------------- /tests/keras/test_gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/keras/test_gcn.py -------------------------------------------------------------------------------- /tests/keras/test_sbcnm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/keras/test_sbcnm.py -------------------------------------------------------------------------------- /tests/keras/test_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/keras/test_transformer.py -------------------------------------------------------------------------------- /tests/keras/test_xdeepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/keras/test_xdeepfm.py -------------------------------------------------------------------------------- /tests/testing.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongmaoTeamTf/deep_recommenders/HEAD/tests/testing.sh --------------------------------------------------------------------------------