Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tfma.extractors.Filter

tfma.extractors.Filter(
    *args,
    **kwargs
)

Filters extracts to include/exclude specified keys.

Args:

  • extracts: PCollection of extracts.
  • include: Keys to include in output.
  • exclude: Keys to exclude from output.

Returns:

Filtered PCollection of Extracts.

Raises:

  • ValueError: If both include and exclude are used.