apache_beam.ml.transforms package

Subpackages

• apache_beam.ml.transforms.embeddings package
  • Submodules
    • apache_beam.ml.transforms.embeddings.huggingface module
      • SentenceTransformerEmbeddings
      • InferenceAPIEmbeddings
    • apache_beam.ml.transforms.embeddings.open_ai module
      • OpenAITextEmbeddings
    • apache_beam.ml.transforms.embeddings.tensorflow_hub module
      • TensorflowHubTextEmbeddings
      • TensorflowHubImageEmbeddings
    • apache_beam.ml.transforms.embeddings.vertex_ai module
      • VertexAITextEmbeddings
      • VertexAIImageEmbeddings

Submodules

• apache_beam.ml.transforms.base module
  • MLTransform
    • MLTransform.expand()
    • MLTransform.with_transform()
    • MLTransform.with_exception_handling()
  • ProcessHandler
    • ProcessHandler.append_transform()
  • MLTransformProvider
    • MLTransformProvider.get_ptransform_for_processing()
    • MLTransformProvider.get_counter()
  • BaseOperation
    • BaseOperation.apply_transform()
  • EmbeddingsManager
    • EmbeddingsManager.get_model_handler()
    • EmbeddingsManager.get_columns_to_apply()
• apache_beam.ml.transforms.handlers module
  • TFTProcessHandler
    • TFTProcessHandler.append_transform()
    • TFTProcessHandler.get_raw_data_feature_spec()
    • TFTProcessHandler.convert_raw_data_feature_spec_to_dataset_metadata()
    • TFTProcessHandler.get_raw_data_metadata()
    • TFTProcessHandler.write_transform_artifacts()
    • TFTProcessHandler.process_data_fn()
    • TFTProcessHandler.expand()
    • TFTProcessHandler.with_exception_handling()
• apache_beam.ml.transforms.tft module
  • ComputeAndApplyVocabulary
    • ComputeAndApplyVocabulary.apply_transform()
  • ScaleToZScore
    • ScaleToZScore.apply_transform()
  • ScaleTo01
    • ScaleTo01.apply_transform()
  • ScaleToGaussian
    • ScaleToGaussian.apply_transform()
  • ApplyBuckets
    • ApplyBuckets.apply_transform()
  • ApplyBucketsWithInterpolation
    • ApplyBucketsWithInterpolation.apply_transform()
  • Bucketize
    • Bucketize.apply_transform()
  • TFIDF
    • TFIDF.apply_transform()
  • TFTOperation
    • TFTOperation.get_ptransform_for_processing()
  • ScaleByMinMax
    • ScaleByMinMax.apply_transform()
  • NGrams
    • NGrams.apply_transform()
  • BagOfWords
    • BagOfWords.apply_transform()
  • HashStrings
    • HashStrings.apply_transform()
  • DeduplicateTensorPerRow
    • DeduplicateTensorPerRow.apply_transform()
• apache_beam.ml.transforms.utils module
  • ArtifactsFetcher
    • ArtifactsFetcher.get_vocab_list()
    • ArtifactsFetcher.get_vocab_filepath()
    • ArtifactsFetcher.get_vocab_size()