tensorflow从入门到精通100讲（三）-谈谈Estimator在Tensorflow中的应用

前言

正如Tensorflow的官网所示：TensorFlow 提供一个包含多个 API 层的编程堆栈其架构图如下。用户可以任意选择不同级别的API进行自己模型的构建。而本篇文章就最高级别的API——Estimator进行一个简短的介绍。

tensorflow API层的编程堆栈

使用Estimator的流程大致如下：

 

• 构建输入数据流
• 定义特征列
• 创建Estimator实例
• 训练模型，进行预测

创建输入数据流

tf.data API 构建输入数据流的流程如下图所示。

 

构建输入数据

 

大致分为三部：

• 从数据源读入数据，api有from_tensor_slices，from_generator等。
• 通过tf.data.Dataset中的函数处理读入数据，api有shuffle, repeat,batch等。