TensorFlow中的API可以用下图表示
Estimator类属于高级 API 。主要用来对模型进行操作。
Datasets类属于中级API。Datasets类主要用于载入 操作 管理数据。
这两类属于在当前版本中常用的重要的类。
1、Estimator:
本文主要根据MNIST例子里用到的来介绍。
模块:
exports:输出Estimator
input:创建输入方法,有两个方法。numpy_input_fn(),pandas_input_fn().链接是官方文档。
#返回一个方法,根据数组字典返回featur和targe.feature的key和x的一样,如果y是个字典,target的Key和y的一样
numpy_input_fn(
x,
y=None,#数组或者字典数组
batch_size=128,
num_epochs=1,#数据迭代次数 If none will run forever.
shuffle=None,#预测时不要用
queue_capacity=1000,#累积队列大小
num_threads=1 #在预测和评估时,用1.
)
2)Estimator类:Estimator类是用来对模型进行训练和评估。