我正在编写一个 API,它配置一个 tfma EvalConfig protobuf 对象并将其发送到下游以在 TFX 管道中使用。目前此 API 运行
from tensorflow_model_analysis import EvalConfig
它初始化了整个 tensorflow 框架。它使我非常小的 API 依赖于整个(巨大的)Tensorflow 框架,并且也不必要地使其依赖于在 tensorflow 实际启动的架构上运行。由于这些原因,我希望能够只导入/加载 protobuf 类。
我已经知道我可以通过在我的项目中硬拷贝整个 .proto 来实现这一点,但我发现这既肮脏又令人反感,如果/当 TFMA 项目决定更改 proto 文件时,它肯定会增加开销。
有没有一种干净的方法可以只从 TFMA 导入 proto 文件?或者来自 tensorflow 的其他 proto 文件?