tensorflow学习笔记（1）- tf.estimator.EstimatorSpec源码分析

tf.estimator.EstimatorSpec源码分析

tf.estimator.EstimatorSpec是python新增的一个高级api，它定义了一个具体的模型对象，现根据源代码简要分析其功能。

ModeKeys类

源代码如下：

class ModeKeys(object):
TRAIN = 'train'
EVAL = 'eval'
PREDICT = 'infer'


该类有点类似于Java中的枚举类型，定义了tensorflow的三种模式：训练、预测、评估模型

EstimatorSpec类

该类定义了一个具体的模型：

def __new__(cls,
mode,
predictions=None,
loss=None,
train_op=None,
eval_metric_ops=None,
export_outputs=None,
training_chief_hooks=None,
training_hooks=None,
scaffold=None):

该类定义了一个具体的模型对象，但是参数之间互相有约束
mode: 定义