第五章:新版TensorFlow入门(自定义模型)
发布时间
阅读量:
阅读量
1、概述
最新版本的tensorflow集成了六种模型,其中包括线性回归模型以及逻辑回归模型,这些模型能够帮助用户高效且便捷地完成各类业务需求的实现。然而,当内置的模型无法满足实际应用需求时,就需要用户自行开发符合特定要求的模型。
如图所示,预先构建的 Estimator 是 tf.estimator.Estimator 基类的一个子类,
而用户自定义的 Estimator 则是 tf.estimator.Estimator 的一个实例:

模型函数(即 model_fn)用于实现机器学习算法。使用预定义 Estimator 与自定义 Estimator 的主要差异在于:
- 当选用预定义 Estimator 时,模型函数的编写工作已由他人完成。
- 若选择自定义 Estimator,则需要用户自行完成模型函数的编写。
所编写的模型函数能够支持多种算法,并可设定各类隐藏层结构及评估指标。与输入函数类似,所有模型函数都需
全部评论 (0)
还没有任何评论哟~
