Skip to content

Post new data to a deployed SageMaker model endpoint and augment with predictions

Usage

# S3 method for vetiver_endpoint_sagemaker
augment(x, new_data, ...)

Arguments

x

A SageMaker model endpoint object created with vetiver_endpoint_sagemaker().

new_data

New data for making predictions, such as a data frame.

...

Extra arguments passed to paws.machine.learning::sagemakerruntime_invoke_endpoint()

Value

The new_data with added prediction column(s).

Examples


if (FALSE) {
  endpoint <- vetiver_endpoint_sagemaker("sagemaker-demo-model")
  augment(endpoint, mtcars[4:7, -1])
}