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