def display_tfma_analysis()

in tfx_airflow/notebooks/tfx_utils.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

  def display_tfma_analysis(self, model_id, slicing_column=None):
    """Displays TFMA metrics for `model_id` sliced by `slicing_column`.

    Args:
      model_id: A `int` indicating the id of a `TFXArtifactTypes.MODEL` artifact
      slicing_column: (Optional) A `str` indicating the slicing column for
          the TFMA metrics.

    Returns:
      A SlicingMetricsViewer object if in Jupyter notebook; None if in Colab.
    """
    tfma_artifact = self.get_dest_artifact_of_type(
        model_id, TFXArtifactTypes.MODEL_EVAL)
    if tfma_artifact:
      return tfma.view.render_slicing_metrics(
          tfma.load_eval_result(tfma_artifact.uri),
          slicing_column=slicing_column)