Keras和Estimator类似,都属于TensorFlow高阶API,提供了方便的构图功能,并对训练、评估、验证、导出提供了方便的接口。 TensorFlow 2.6版本中继续支持Keras API,如果需要沿用在TensorFlow1.x版本中的用法,则可以通过compat.v1模块调用,调用方式如下: 1 tf.compat.v1.Session 使用Keras API进行训练脚本开发的一般步骤为: 数据预处理。 模型搭建。 模型编译。 模型训练。 当前仅支持通过TensorFlow的Keras API编写的训练脚本,而不支持原生Keras API。 下面介绍如何迁移此类Keras训练