Uploaded image for project: 'Beam'
  1. Beam
  2. BEAM-9145 Integrate Google Cloud AI functionalities
  3. BEAM-9501

[Java] PTransform that integrates Cloud AI Prediction functionality

Details

    • Sub-task
    • Status: Open
    • P3
    • Resolution: Unresolved
    • None
    • None
    • io-java-gcp
    • None

    Description

      The goal is to create a PTransform that integrates Google Cloud AI Platform Prediction functionality [1] for Java SDK. The transform should work both in batch and streaming mode.

      One possible solution is to utilize a `tfx_bsl` library [2] which offers a `RunInference` Beam's PTransform[2] able to get predictions from Google Cloud AI Platform Prediction via cross-language. 

      [1] https://cloud.google.com/ai-platform/prediction/docs/
      [2] https://github.com/tensorflow/tfx-bsl/blob/ecec31b2152fc73f4c26f5a5db86173cfcf25c99/tfx_bsl/beam/run_inference.py#L99

       

      Attachments

        Activity

          People

            Unassigned Unassigned
            kamilwu Kamil Wasilewski
            Votes:
            0 Vote for this issue
            Watchers:
            3 Start watching this issue

            Dates

              Created:
              Updated: