# SageMaker Scikit-learn Extension Container

The SageMaker Scikit-learn Extension Container is used in SageMaker Autopilot.

The SageMaker Scikit-learn Extension Container is built in 3 steps. The first 2 steps should be the same as building the [sagemaker-scikit-learn-container](https://github.com/aws/sagemaker-scikit-learn-container) image.

### Step 1: Base Image

The "base" Dockerfile encompass the installation of the framework and all of the dependencies needed.

Tagging scheme is based on <Scikit-learn_version>-<SageMaker_version>-cpu-py<python_version>. (e.g. 1.2-1-cpu-py3)

All "final" Dockerfiles build images using base images that use the tagging scheme above.

```
docker build -t sklearn-base:1.2-1-cpu-py3 -f docker/1.2-1/base/Dockerfile.cpu .
```

Notice that this Dockerfile has the updated version of sklearn (1.2.1) installed.

### Step 2: Final Image

The "final" Dockerfiles encompass the installation of the SageMaker specific support code.

All "final" Dockerfiles use base images for building.

These "base" images are specified with the naming convention of sklearn-base:<Scikit-learn_version>-<SageMaker_version>-cpu-py<python_version>.

Before building "final" images:

Build your "base" image. Make sure it is named and tagged in accordance with your "final" Dockerfile.

```
# Create the SageMaker Scikit-learn Container Python package.
python setup.py bdist_wheel
```

Then build the final image, like in the sagemaker-sklearn-container

```
docker build -t preprod-sklearn:1.2-1-cpu-py3 -f docker/1.2-1/final/Dockerfile.cpu .
```

### Step 3: Build the extension image for SageMaker Scikit-learn Extension Container

The "extension" Dockerfiles encompass the installation of the SageMaker Autopilot specific support code.

The "extension" Dockerfiles use final images for building.

Build the third additional Dockerfile needed for SageMaker Scikit-learn Extension Container. This Dockerfile specifies a hard dependency on a certain version of scikit-learn (i.e. v1.2.1).

Tagging scheme is based on extension-<Scikit-learn-Extension_version>-<SageMaker_version>-cpu-py<python_version>. (e.g. extension-2.5-1-cpu-py3). Make sure the "extension" image is tagged in accordance with the  `extension` (i.e. `extension-2.5-1-cpu-py3`).

```
docker build -t preprod-sklearn-extension:2.5-1-cpu-py3 -f  docker/1.2-1/extension/Dockerfile.cpu .
```