TensorFlow Model Analysis 0.32.1模型评估工具

模型评估的自动化工具包,tensorflow_model_analysis 是我最近挺喜欢用的一个库。适合你在用 TFX 或做 ML pipeline 时,需要批量评估模型表现的时候。它不仅能跑各种 slice 的指标,还能自动出一堆直观的可视化,挺方便的。

搭配 TensorFlow Extended 一起用更爽,pipeline 一跑完,评估结果图表就全都出来了。你要看 AUC、precision、recall 这些,通通搞定。响应也快,配合 tfdvtfma.EvalResult,整个流程都能打通。

部署也比较简单,tensorflow_model_analysis-0.32.1-py3-none-any.whl 就是个 .whl 包,直接用 pip install 装一下就能跑。没啥坑,就是版本别下错了,对 TensorFlow 版本有点依赖,建议配套用。

如果你正在做模型上线前的评估验证,这个库真挺香的。是你用 Jupyter Notebook 的话,还能直接输出交互图,滑条拖一拖,体验不错。

想了解更多数据和可视化的内容,可以看看这些链接,像是 Python 数据与可视化实战案例基于 Pandas 的时间序列数据 也都挺有用的。

whl 文件大小:1.63MB