tfma.extractors.Extractor

Extractor(stage_name, ptransform)

stage_name A namedtuple alias for field number 0
ptransform A namedtuple alias for field number 1