TensorFlow入门10 -- 编程堆栈(programming stack)

参考:《深度学习图像识别技术--基于TensorFlow Object Detection API 和 OpenVINO

TensorFlow提供了一个由多个API层组成的编程堆栈(programming stack),如下图所示:


TensorFlow官网强烈推荐用下面两个APIs编写TensorFlow程序

Estimators, Estimator API 提供了完整的训练模型、评估模型精度和生成预测的方法(methods) 

Datasets, 构建数据输入管道(data input pipeline). Dataset API 负责载入数据、操作数据,然后把数据输入模型 。Dataset API 和 Estimators API 啮合的非常好。

获得范例代码

用命令(若没有装git,git安装和使用请参考《Git学习手记3:安装Git》):

git clone https://github.com/tensorflow/models, 如下


进入models/samples/core/get_started/ 文件夹下,运行premade_estimator.py


运行结果如下:


在Visual Studio Code中,打开文件premade_estimator.py,如下图所示:


可以看到该程序主要分为以下六个部分:

1,导入和解析数据集。

2,创建描述数据的特征列。

3,选择模型的类型。

4,训练模型。

5,评估模型的有效性。

6,让训练好的模型进行预测。

这六个部分也是大多数TensorFlow程序包含的必要部分

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容