apache_beam.ml.transforms.embeddings package

This module contains embedding configs that can be used to generate embeddings using MLTransform.

Submodules

• apache_beam.ml.transforms.embeddings.huggingface module
  • SentenceTransformerEmbeddings
    • SentenceTransformerEmbeddings.get_model_handler()
    • SentenceTransformerEmbeddings.get_ptransform_for_processing()
  • InferenceAPIEmbeddings
    • InferenceAPIEmbeddings.get_token()
    • InferenceAPIEmbeddings.api_url
    • InferenceAPIEmbeddings.authorization_token
    • InferenceAPIEmbeddings.get_model_handler()
    • InferenceAPIEmbeddings.get_ptransform_for_processing()
• apache_beam.ml.transforms.embeddings.tensorflow_hub module
  • TensorflowHubTextEmbeddings
    • TensorflowHubTextEmbeddings.get_model_handler()
    • TensorflowHubTextEmbeddings.get_ptransform_for_processing()
  • TensorflowHubImageEmbeddings
    • TensorflowHubImageEmbeddings.get_model_handler()
    • TensorflowHubImageEmbeddings.get_ptransform_for_processing()
• apache_beam.ml.transforms.embeddings.vertex_ai module
  • VertexAITextEmbeddings
    • VertexAITextEmbeddings.get_model_handler()
    • VertexAITextEmbeddings.get_ptransform_for_processing()
  • VertexAIImageEmbeddings
    • VertexAIImageEmbeddings.get_model_handler()
    • VertexAIImageEmbeddings.get_ptransform_for_processing()