AnandInguva commented on code in PR #21801:
URL: https://github.com/apache/beam/pull/21801#discussion_r895961267
##########
sdks/python/apache_beam/ml/inference/base.py:
##########
@@ -206,6 +216,19 @@ class
RunInference(beam.PTransform[beam.PCollection[ExampleT],
Args:
model_handler: An implementation of ModelHandler.
clock: A clock implementing get_current_time_in_microseconds.
+
+ A transform that takes a PCollection of examples (or features) to be used on
+ an ML model. It will then output inferences (or predictions) for those
+ examples in a PCollection of PredictionResults, containing the input examples
+ and output inferences.
+
+ If examples are paired with keys, it will output a tuple
+ (key, PredictionResult) for each (key, example) input.
+
+ Models for supported frameworks can be loaded via a URI. Supported services
Review Comment:
Yep. thanks
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
