├── .gitignore ├── README.md ├── data ├── __init__.py ├── census-income.data.gz ├── census-income.test.gz └── criteo_sample.txt ├── layer ├── __init__.py ├── common_layer.py └── layers.py ├── metrics ├── __init__.py └── metrics.py ├── model ├── __init__.py ├── afm.py ├── attention.py ├── dcn.py ├── fm.py ├── mmoe.py ├── pnn.py └── wide_and_deep.py ├── run_afm_on_criteo_sample.py ├── run_attention_on_criteo_sample.py ├── run_dcn_on_criteo_sample.py ├── run_fm_on_criteo_sample.py ├── run_mmoe_on_census_income.py ├── run_pnn_on_criteo_sample.py ├── run_wide_and_deep_on_criteo_sample.py ├── structs ├── __init__.py └── input_structs.py └── util ├── __init__.py ├── input_util.py └── model_util.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea 2 | tmp 3 | test 4 | .data 5 | save_model 6 | delete 7 | paper -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/census-income.data.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/data/census-income.data.gz -------------------------------------------------------------------------------- /data/census-income.test.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/data/census-income.test.gz -------------------------------------------------------------------------------- /data/criteo_sample.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/data/criteo_sample.txt -------------------------------------------------------------------------------- /layer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/layer/__init__.py -------------------------------------------------------------------------------- /layer/common_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/layer/common_layer.py -------------------------------------------------------------------------------- /layer/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/layer/layers.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/metrics/__init__.py -------------------------------------------------------------------------------- /metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/metrics/metrics.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/afm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/model/afm.py -------------------------------------------------------------------------------- /model/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/model/attention.py -------------------------------------------------------------------------------- /model/dcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/model/dcn.py -------------------------------------------------------------------------------- /model/fm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/model/fm.py -------------------------------------------------------------------------------- /model/mmoe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/model/mmoe.py -------------------------------------------------------------------------------- /model/pnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/model/pnn.py -------------------------------------------------------------------------------- /model/wide_and_deep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/model/wide_and_deep.py -------------------------------------------------------------------------------- /run_afm_on_criteo_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/run_afm_on_criteo_sample.py -------------------------------------------------------------------------------- /run_attention_on_criteo_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/run_attention_on_criteo_sample.py -------------------------------------------------------------------------------- /run_dcn_on_criteo_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/run_dcn_on_criteo_sample.py -------------------------------------------------------------------------------- /run_fm_on_criteo_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/run_fm_on_criteo_sample.py -------------------------------------------------------------------------------- /run_mmoe_on_census_income.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/run_mmoe_on_census_income.py -------------------------------------------------------------------------------- /run_pnn_on_criteo_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/run_pnn_on_criteo_sample.py -------------------------------------------------------------------------------- /run_wide_and_deep_on_criteo_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/run_wide_and_deep_on_criteo_sample.py -------------------------------------------------------------------------------- /structs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/structs/__init__.py -------------------------------------------------------------------------------- /structs/input_structs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/structs/input_structs.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/util/__init__.py -------------------------------------------------------------------------------- /util/input_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/util/input_util.py -------------------------------------------------------------------------------- /util/model_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yezhiwen/tensorflow_recommend_algorithm/HEAD/util/model_util.py --------------------------------------------------------------------------------