sagemaker.core.lineage.visualizer

sagemaker.core.lineage.visualizer#

This module contains functionality to display lineage data.

Classes

LineageTableVisualizer(sagemaker_session)

Creates a dataframe containing the lineage assoociations of a SageMaker object.
class sagemaker.core.lineage.visualizer.LineageTableVisualizer(sagemaker_session)[source]#

Bases: object

Creates a dataframe containing the lineage assoociations of a SageMaker object.

show(trial_component_name: str | None = None, training_job_name: str | None = None, processing_job_name: str | None = None, pipeline_execution_step: object | None = None, model_package_arn: str | None = None, endpoint_arn: str | None = None, artifact_arn: str | None = None, context_arn: str | None = None, actions_arn: str | None = None) → DataFrame[source]#

Generate a dataframe containing all incoming and outgoing lineage entities.

Examples: .. code-block:: python

viz = LineageTableVisualizer(sagemaker_session) df = viz.show(training_job_name=training_job_name) # in a notebook display(df.to_html())

Parameters:

  • trial_component_name (str, optional) – Name of a trial component. Defaults to None.

  • training_job_name (str, optional) – Name of a training job. Defaults to None.

  • processing_job_name (str, optional) – Name of a processing job. Defaults to None.

  • pipeline_execution_step (obj, optional) – Pipeline execution step. Defaults to None.

  • model_package_arn (str, optional) – Model package arn. Defaults to None.

  • endpoint_arn (str, optional) – Endpoint arn. Defaults to None.

  • artifact_arn (str, optional) – Artifact arn. Defaults to None.

  • context_arn (str, optional) – Context arn. Defaults to None.

  • actions_arn (str, optional) – Action arn. Defaults to None.

Returns:

Pandas dataframe containing lineage associations.

Return type:

DataFrame

Contents