syne_tune.backend.sagemaker_backend.custom_framework module

class syne_tune.backend.sagemaker_backend.custom_framework.CustomFramework(entry_point, image_uri, source_dir=None, hyperparameters=None, **kwargs)

Bases: Framework

LATEST_VERSION = '0.1'

create_model(model_server_workers=None, role=None, vpc_config_override='VPC_CONFIG_DEFAULT')

Create a SageMaker Model object that can be deployed to an Endpoint.

Args:

**kwargs: Keyword arguments used by the implemented method for

creating the Model.

Returns:

sagemaker.model.Model: A SageMaker Model object. See Model() for full details.