ray.train.batch_predictor.BatchPredictor.from_pandas_udf#

classmethod BatchPredictor.from_pandas_udf(pandas_udf: Callable[[pandas.core.frame.DataFrame], pandas.core.frame.DataFrame]) → ray.train.batch_predictor.BatchPredictor[source]#

Create a Predictor from a Pandas UDF.

Parameters

pandas_udf – A function that takes a pandas.DataFrame and other optional kwargs and returns a pandas.DataFrame.