易百教程

48、TensorFlow Estimators是什么?

Estimators 是一个高级 API,它减少了之前在训练 TensorFlow 模型时需要编写的大部分代码。估计器非常灵活,如果模型有任何特定要求,它允许覆盖默认行为。

使用 Estimator 构建模型有两种可能的方法:

预制估算器
这些是预定义的估计器。创建这些模型是为了生成特定类型的模型,例如 DNNClassifier,它是一个预制的估计器。
估计器(基类)
它提供了对如何使用 model_fn 函数创建模型的完全控制。
这些函数由类 tf.estimator.Estimator 使用。这些函数返回一个初始化的估计器,我们可以在其上调用 .train.eval.predict
Estimator 的示意图可以看作: