如何在大型数据集的特定切片上评估TensorFlow模型的性能?

时间:2018-03-29 22:59:08

标签: tensorflow tensorflow-model-analysis

如何评估TensorFlow模型在大型评估数据集的特定切片(片段)上的性能?

1 个答案:

答案 0 :(得分:2)

使用TensorFlow模型分析(TFMA),这是一个开源库,结合TensorFlow和Apache Beam来计算和可视化评估指标。它专为此用例而设计,允许您使用TensorFlow培训师中定义的相同指标,以分布式方式评估大量数据的模型。这些指标也可以在不同的数据切片上计算,结果可以在Jupyter笔记本中显示。 TFMA使用Apache Beam对指定的评估数据集执行完整传递。这不仅可以更准确地计算指标,还可以扩展到大量评估数据集,因为可以使用分布式处理后端运行Beam管道。

有关详细信息,请参阅https://github.com/tensorflow/model-analysis