TensorFlow学习笔记-tf.estimator

tf.estimator.Estimator

Estimator class训练和测试TF模型。Estimator对象封装好通过model_fn指定的模型，给定输入和其它超参数，返回ops执行training, evaluation or prediction. 所有的输出(包含checkpoints, event files, etc.)被写入model_dir。

属性

• config
  

  传入 model_fn,如果 model_fn有参数named “config”

• model_dir
• model_fn
  The model_fn with following signature: def model_fn(features, labels, mode, config)
• params

方法

• __init__

__init__(
    model_fn,
    model_dir=None,
    config=None,
    params=None # 将要传入model_fn的超参数字典
)

• <