SKLearnModel: SKLearnModel Class

In DyfanJones/sagemaker-r-mlframework: sagemaker machine learning developed by amazon

SKLearnModel Class

Description

An Scikit-learn SageMaker “Model“ that can be deployed to a SageMaker “Endpoint“.

Super classes

sagemaker.mlcore::ModelBase -> sagemaker.mlcore::Model -> sagemaker.mlcore::FrameworkModel -> SKLearnModel

Methods

Public methods

• SKLearnModel$new()

• SKLearnModel$prepare_container_def()

• SKLearnModel$serving_image_uri()

• SKLearnModel$clone()

Inherited methods

• sagemaker.mlcore::ModelBase$format()

• sagemaker.mlcore::ModelBase$help()

• sagemaker.mlcore::Model$.create_sagemaker_model()

• sagemaker.mlcore::Model$check_neo_region()

• sagemaker.mlcore::Model$compile()

• sagemaker.mlcore::Model$delete_model()

• sagemaker.mlcore::Model$deploy()

• sagemaker.mlcore::Model$enable_network_isolation()

• sagemaker.mlcore::Model$package_for_edge()

• sagemaker.mlcore::Model$register()

• sagemaker.mlcore::Model$transformer()

---

Method new()

Initialize an SKLearnModel.

Usage

SKLearnModel$new(
  model_data,
  role,
  entry_point,
  framework_version = NULL,
  py_version = "py3",
  image_uri = NULL,
  predictor_cls = SKLearnPredictor,
  model_server_workers = NULL,
  ...
)

Arguments

model_data

(str): The S3 location of a SageMaker model data “.tar.gz“ file.

role

(str): An AWS IAM role (either name or full ARN). The Amazon SageMaker training jobs and APIs that create Amazon SageMaker endpoints use this role to access training data and model artifacts. After the endpoint is created, the inference code might use the IAM role, if it needs to access an AWS resource.

entry_point

(str): Path (absolute or relative) to the Python source file which should be executed as the entry point to model hosting. If “source_dir“ is specified, then “entry_point“ must point to a file located at the root of “source_dir“.

framework_version

(str): Scikit-learn version you want to use for executing your model training code. Defaults to “None“. Required unless “image_uri“ is provided.

py_version

(str): Python version you want to use for executing your model training code (default: 'py3'). Currently, 'py3' is the only supported version. If “None“ is passed in, “image_uri“ must be provided.

image_uri

(str): A Docker image URI (default: None). If not specified, a default image for Scikit-learn will be used. If “framework_version“ or “py_version“ are “None“, then “image_uri“ is required. If also “None“, then a “ValueError“ will be raised.

predictor_cls

(callable[str, sagemaker.session.Session]): A function to call to create a predictor with an endpoint name and SageMaker “Session“. If specified, “deploy()“ returns the result of invoking this function on the created endpoint name.

model_server_workers

(int): Optional. The number of worker processes used by the inference server. If None, server will use one worker per vCPU.

...

: Keyword arguments passed to the “FrameworkModel“ initializer.

---

Method prepare_container_def()

Return a container definition with framework configuration set in model environment variables.

Usage

SKLearnModel$prepare_container_def(
  instance_type = NULL,
  accelerator_type = NULL
)

Arguments

instance_type

(str): The EC2 instance type to deploy this Model to. This parameter is unused because Scikit-learn supports only CPU.

accelerator_type

(str): The Elastic Inference accelerator type to deploy to the instance for loading and making inferences to the model. This parameter is unused because accelerator types are not supported by SKLearnModel.

Returns

dict[str, str]: A container definition object usable with the CreateModel API.

---

Method serving_image_uri()

Create a URI for the serving image.

Usage

SKLearnModel$serving_image_uri(region_name, instance_type)

Arguments

region_name

(str): AWS region where the image is uploaded.

instance_type

(str): SageMaker instance type.

Returns

str: The appropriate image URI based on the given parameters.

---

Method clone()

The objects of this class are cloneable with this method.

Usage

SKLearnModel$clone(deep = FALSE)

Arguments

deep

Whether to make a deep clone.

DyfanJones/sagemaker-r-mlframework documentation built on March 18, 2022, 7:41 a.m.