def compare_examples_stats_for_models()

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

• 10 lines of code
• 3 McCabe index (conditional complexity)

  def compare_examples_stats_for_models(self, model_id, other_model_id):
    """Compares stats for examples to train `model_id` & `other_model_id`."""
    examples_artifact, other_examples_artifact = (
        self.get_source_artifact_of_type(model_id, TFXArtifactTypes.EXAMPLES),
        self.get_source_artifact_of_type(
            other_model_id, TFXArtifactTypes.EXAMPLES)
    )
    if examples_artifact and other_examples_artifact:
      self.compare_stats_for_examples(
          examples_artifact.id, other_examples_artifact.id,
          name='model_'+str(model_id), other_name='model_'+str(other_model_id))