Andy Ye created BEAM-13985:
------------------------------

             Summary: Implement end-to-end tests for RunInference classes
                 Key: BEAM-13985
                 URL: https://issues.apache.org/jira/browse/BEAM-13985
             Project: Beam
          Issue Type: Sub-task
          Components: sdk-py-core
            Reporter: Andy Ye
            Assignee: Anand Inguva


Make calls to the RunInference classes for TFX, Pytorch, and Scikit-learn.
 * For TFX, need to use their proto

Process
 * Read data from GCS bucket
 * Use trained model? or simple?
 * Product the output predictions
 * Assert if output predictions match actual

Add task for using GPU container images



--
This message was sent by Atlassian Jira
(v8.20.1#820001)

Reply via email to