Advertisement

第五章:新版TensorFlow入门(自定义模型)

阅读量:

1、概述

最新版本的tensorflow集成了六种模型,其中包括线性回归模型以及逻辑回归模型,这些模型能够帮助用户高效且便捷地完成各类业务需求的实现。然而,当内置的模型无法满足实际应用需求时,就需要用户自行开发符合特定要求的模型。

如图所示,预先构建的 Estimator 是 tf.estimator.Estimator 基类的一个子类,

而用户自定义的 Estimator 则是 tf.estimator.Estimator 的一个实例:

模型函数(即 model_fn)用于实现机器学习算法。使用预定义 Estimator 与自定义 Estimator 的主要差异在于:

  • 当选用预定义 Estimator 时,模型函数的编写工作已由他人完成。
  • 若选择自定义 Estimator,则需要用户自行完成模型函数的编写。

所编写的模型函数能够支持多种算法,并可设定各类隐藏层结构及评估指标。与输入函数类似,所有模型函数都需

全部评论 (0)

还没有任何评论哟~