常用的深度学习模型训练、评估和预测相关代码,基于Tensorflow高阶API(Estimator)实现;尽量做到可读性和通用性较好。
部分模型子目录下有较详细的文档介绍!
关于其中某些模型的介绍,请参考文章《主流CTR预估模型的演化及对比》!《深度CTR预估模型中的特征自动组合机制演化简史》
除非特殊说明,本项目的代码都是基于tensorflow 1.6.0开发。
推荐阅读:
- 基于Tensorflow高阶API构建大规模分布式深度学习模型系列: 开篇
- 基于Tensorflow高阶API构建大规模分布式深度学习模型系列:基于Dataset API处理Input pipeline
- 基于Tensorflow高阶API构建大规模分布式深度学习模型系列: 自定义Estimator(以文本分类CNN模型为例)
- 基于Tensorflow高阶API构建大规模分布式深度学习模型系列:特征工程 Feature Column
- 基于Tensorflow高阶API构建大规模分布式深度学习模型系列:CVR预估案例之ESMM模型
- 欢迎关注我的知乎专栏:算法工程师的自我修养
- 欢迎收藏我的个人博客,会不定期更新:https://yangxudong.github.io, 或者国内镜像:https://xudongyang.coding.me