.png%3Falt%3Dmedia%26token%3D3f1e92da-c9f8-4b18-a9ca-5f0c5faa81ad)
.png%3Falt%3Dmedia%26token%3D3f1e92da-c9f8-4b18-a9ca-5f0c5faa81ad)
Visualize artifacts
Configuring ZenML to display data visualizations in the dashboard.
ZenML automatically saves visualizations of many common data types and allows you to view these visualizations in the ZenML dashboard:
ZenML Artifact Visualizations
Alternatively, any of these visualizations can also be displayed in Jupyter notebooks using the visualize post-execution method:
output.visualize() Output
Currently, the following visualization types are supported:
- HTML: Embedded HTML visualizations such as data validation reports,
- Image: Visualizations of image data such as Pillow images or certain numeric numpy arrays,
- CSV: Tables, such as the pandas DataFrame
.describe()output, - Markdown: Markdown strings or pages.
There are two ways how you can add custom visualizations to the dashboard:
- If you are already handling HTML, Markdown, or CSV data in one of your steps, you can have them visualized in just a few lines of code by casting them to a special class inside your step.
- If you want to automatically extract visualizations for all artifacts of a certain data type, you can define type-specific visualization logic by building a custom materializer.
If you already have HTML, Markdown, or CSV data available as a string inside your step, you can simply cast them to one of the following types and return them from your step:
zenml.types.HTMLStringfor strings in HTML format, e.g.,"# Header\nSome text",zenml.types.MarkdownStringfor strings in Markdown format, e.g.,"<h1>Header</h1>Some text",zenml.types.CSVStringfor strings in CSV format, e.g.,"a,b,c\n1,2,3".
from zenml.types import CSVString
@step
def my_step() -> CSVString:
some_csv = "a,b,c\n1,2,3"
return CSVString(some_csv)
This would create the following visualization in the dashboard:
CSV Visualization Example
If you want to automatically extract visualizations for all artifacts of a certain data type, you can do so by overriding the
save_visualizations method of the corresponding materializer. See the materializer docs page for more information on how to create custom materializers that do this.If you would like to disable artifact visualization altogether, you can set
enable_artifact_visualization at either pipeline or step level:@step(enable_artifact_visualization=False)
def my_step():
...
@pipeline(enable_artifact_visualization=False)
def my_pipeline():
...
If all visualizations of a certain pipeline run are not showing up in the dashboard, it might be that your ZenML server does not have the required dependencies or permissions to access that artifact store. See the custom artifact store docs page for more information.
Last modified 2d ago