apache_beam.ml.transforms.utils module¶

class apache_beam.ml.transforms.utils.ArtifactsFetcher(artifact_location)[source]¶

Bases: object

Utility class used to fetch artifacts from the artifact_location passed to the TFTProcessHandlers in MLTransform.

get_vocab_list(vocab_filename: str = 'compute_and_apply_vocab') → List[bytes][source]¶: Returns list of vocabulary terms created during MLTransform.

get_vocab_filepath(vocab_filename: str = 'compute_and_apply_vocab') → str[source]¶: Return the path to the vocabulary file created during MLTransform.

get_vocab_size(vocab_filename: str = 'compute_and_apply_vocab') → int[source]¶