其他

Github 项目推荐 | TensorFlow 的模型分析工具 —— TFMA

2018-04-05 孔令双 AI研习社

TFMA 是一个用于评估 TensorFlow 模型的库,它可以让用户使用 Trainer 里定义的指标以分布式方式评估大量数据的模型。这些指标也可以在不同的数据片里计算,其结果可以在 Jupyter Notebooks 里可视化。

TFMA 可能会在版本 1.0 之前引入后向不兼容的更改。

Github:

https://github.com/tensorflow/model-analysis

  安装

最方便且最推荐的安装 TFMA 的方法是使用 PyPI 包:

pip install tensorflow-model-analysis

在安装 TFMA 之前需要装好 TensorFlow,但是没必要将 TensorFlow 当作一个明确的依赖包。有关 TensorFlow 的安装方法,请查阅 TF 官方文档。

在 Jupyter Notebooks 里可视化 TFMA,请运行:

jupyter nbextension enable --py widgetsnbextension
jupyter nbextension install --py --symlink tensorflow_model_analysis
jupyter nbextension enable --py tensorflow_model_analysis

TFMA 要求 Apache Beam 运行分布式管道,Apache Beam 默认以本地模式运行,也可以使用 Google Cloud Dataflow 以分布式模式运行。TFMA 可以扩展到其他的 Apache Beam 的 runner 上。

  兼容版本

根据我们的测试框架,这是一个已知互相兼容的版本表。 其他组合也可以工作,但未经测试。

4 月 AI 求职季

8 大明星企业

10 场分享盛宴

20 小时独门秘籍

4.10-4.19,我们准时相约!



新人福利



关注 AI 研习社(okweiwu),回复  1  领取

【超过 1000G 神经网络 / AI / 大数据资料】



阿里将 TVM 融入 TensorFlow,在 GPU 上实现全面提速

▼▼▼

您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存