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())