tensorflow / tfx-addons

Developers helping developers. TFX-Addons is a collection of community projects to build new components, examples, libraries, and tools for TFX. The projects are organized under the auspices of the special interest group, SIG TFX-Addons. Join the group at http://goo.gle/tfx-addons-group

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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?