apache_beam.ml package

Contains packages for supported machine learning transforms.

Subpackages

• apache_beam.ml.gcp package
  • Submodules
    • apache_beam.ml.gcp.cloud_dlp module
    • apache_beam.ml.gcp.naturallanguageml module
    • apache_beam.ml.gcp.recommendations_ai module
    • apache_beam.ml.gcp.videointelligenceml module
    • apache_beam.ml.gcp.visionml module
• apache_beam.ml.inference package
  • Submodules
    • apache_beam.ml.inference.base module
    • apache_beam.ml.inference.huggingface_inference module
    • apache_beam.ml.inference.onnx_inference module
    • apache_beam.ml.inference.pytorch_inference module
    • apache_beam.ml.inference.sklearn_inference module
    • apache_beam.ml.inference.tensorflow_inference module
    • apache_beam.ml.inference.tensorrt_inference module
    • apache_beam.ml.inference.utils module
    • apache_beam.ml.inference.vertex_ai_inference module
    • apache_beam.ml.inference.xgboost_inference module
• apache_beam.ml.transforms package
  • Subpackages
    • apache_beam.ml.transforms.embeddings package
      • Submodules
        • apache_beam.ml.transforms.embeddings.huggingface module
        • apache_beam.ml.transforms.embeddings.tensorflow_hub module
        • apache_beam.ml.transforms.embeddings.vertex_ai module
  • Submodules
    • apache_beam.ml.transforms.base module
    • apache_beam.ml.transforms.handlers module
    • apache_beam.ml.transforms.tft module
    • apache_beam.ml.transforms.utils module