Tensorflow高阶API实现的深度学习模型训练评估和预测相关代码
本项目提供常用的深度学习模型训练、评估和预测相关代码,基于Tensorflow高阶API Estimator实现。为了尽量提高代码的可读性和通用性,我们提供了较为详细的文档介绍。部分模型子目录下还提供了特定模型的介绍,请参考文章“主流CTR预估模型的演化及对比:深度CTR预估模型中的特征自动组合机制演化简史”。除非特殊说明,本项目的代码都是基于tensorflow1.6.0开发。推荐阅读Tensorflow高阶API构建大规模分布式深度学习模型系列文章,包括“基于Dataset API处理Input pipeline”、 “自定义Estimator以文本分类CNN模型为例”、“特征工程Feature Column”以及“CVR预估案例之ESMM模型”。本项目的源代码可以在Github上找到(https://github.com/yangxudong/deep)。
用户评论