def predict_wine_quality_push_to_production(endpoint_name="wineapp"):
from dbnd_examples.pipelines.wine_quality.serving.sagemaker import (
deploy_to_sagemaker,
push_image_to_ecr,
)
from dbnd_examples.pipelines.wine_quality.serving import docker, sagemaker
model, validation = predict_wine_quality()
package = docker.package_as_docker(model=model)
push = push_image_to_ecr(image=package, name=endpoint_name)
serving = deploy_to_sagemaker(app_name=endpoint_name,
image_url=push,
model_path=model)
test = sagemaker.test_sagemaker_endpoint.t(endpoint_name=endpoint_name)
test.set_upstream(serving)
return serving, test
def predict_wine_quality_package():
model, validation = predict_wine_quality()
serving = package_as_docker(model=model)
return model, validation, serving
def tran_model_and_package():
model, validation = train_model_for_customer()
serving = package_as_docker(model=model)
return model, validation, serving