Skip to main content

Predict

Deployment.predict

Deployment.predict(input_df, stream)

Make predictions for each row of the pandas dataframe.

Parameters:

   input_df: pd.DataFrame
Input pandas dataframe.

   stream: bool
Optional flag to stream results via gRPC protocol (default False).

Returns:

   pd.DataFrame pandas dataframe with predictions.

Examples:

Get the taxi deployment, and make predictions over gRPC using test dataset.

    df = pd.read_csv('titanic_test.csv')

deployment = pc.list_deployments('taxi_deployment')[0]

df = deployment.predict(df, stream=True)
df.head()

Deployment.async_predict

Deployment.async_predict(input_df, stream)

Asyncio version of Deployment.predict.

Parameters:

   input_df: pd.DataFrame
Input pandas dataframe.

   stream: bool
Optional flag to stream results via gRPC protocol (default False).

Returns:

   pd.DataFrame pandas dataframe with predictions.

Examples:

Make predictions on two deployments in parallel using asyncio:

import asyncio
import pandas as pd

async def main():
deployment_1 = pc.list_deployments('taxi_deployment')[0]
deployment_2 = pc.list_deployments('taxi_deployment')[1]

df = pd.read_csv("/path/to/file.csv")

future = await asyncio.gather(
deployment_1.async_predict(df, stream=True),
deployment_2.async_predict(df, stream=True),
)

print(future[0].head(3))
print(future[1].head(3))

if __name__ == "__main__":
asyncio.run(main())