ML Lineage UI for TFX
theadactyl opened this issue · comments
While all the data exists in MLMD, we currently have to rely on orchestrator to represent the executions and other metadata in a nice and organized UI. This would bring a frontend and backend to generate Web UIs from TFX metadata.
cc @casassg
Would love to see this too. In case you haven't come across it there is prior art within the tfx repo, and a frontend built into the kubeflow project
Edit: I guess you have seen the kubeflow UI... Anyway might be useful for anybody else who ends up here
- Similar to KF, Airflow lineage UI but with better support for artifact exploration
- Twitter has an implementation using React, has been useful
- What do people want to see in it? What functionality?
- KF tool has been useful for showing managers the use of lineage
- Should follow the library development
We have a new implementation of a lineage explorer over at ZenML https://github.com/zenml-io/zenml/tree/main/examples/lineage. Would the community be interested in addition this as part of tfx-addons?
@htahir1 that would be great! I think it would be interesting first to have the lineage as a low level API from MLMD (viz can be added after we have that). I'm one of the owners for MLMD Client, so if you want to contribute it there: https://github.com/tensorflow/tfx-addons/tree/main/tfx_addons/mlmd_client I would be happy to review :D
Curious on what you think this could be implemented like?