Skip to content

Click in-app to access the full platform documentation for your version of DataRobot.

Custom model environments

To create a custom inference model, you must select an environment that the model will use. An environment includes packages and language and system libraries used for models. You can select one of two types of environments:

  • Drop-in environments contain web server Scoring Code and a start_server.sh file for the model to use. They are provided by DataRobot in the Workshop.

  • Custom environments do not contain the Scoring Code and start_server.sh file, which instead must be provided in the folder of the custom model you intend to use with the environment. You can create your own in the Workshop. You can also create a custom drop-in environment by including the Scoring Code and start_server.sh file in the environment folder.

By providing an environment separate from a custom model, DataRobot can build the environment for you. This allows you to reuse the environment for as many models as you want. It also provides the ability to provide a model by uploading a folder containing its code and model artifacts without providing web server Scoring Code and a start_server.sh file with every model.

Drop-in environments

DataRobot provides drop-in environments from the Custom Model Workshop. Drop-in environments contain the web server Scoring Code and a start_server.sh file for the custom model so that you do not need to provide them in the model's folder. The following table details the drop-in environments provided by DataRobot. Each environment is prefaced with [DataRobot] in the Environments tab of the Custom Model Workshop.

Environment Name Compatibility
Java Drop-In DataRobot Scoring Code models
Python 3 PyTorch Drop-In PyTorch models (.pth)
Python 3 Scikit-Learn Drop-In Scikit-Learn models (.pkl)
Python 3 XGBoost Drop-In Native XGBoost models (.pkl)
Python 3 Keras Drop-In Keras models backed by tensorflow (.h5)
R Drop-in Environment Models trained using caret if the method name and package name match (check here) (Log in to GitHub before clicking this link.)

Note

All Python environments contain Scikit-Learn to help with preprocessing (if necessary), but only Scikit-Learn can make predictions on sklearn models.

Create a custom environment

To create a custom environment, you must upload a compressed folder in .tar, .tar.gz, or .zip format. Be sure to review the guidelines for preparing a custom environment folder before proceeding. You may also consider creating a drop-in environment by adding Scoring Code and a start_server.sh file to your environment folder.

Environment limits and environment version limits

Next to the Add new environment and the New version buttons, there is a badge indicating how many environments (or environment versions) you've added and how many environments (or environment versions) you can add in total. With the correct permissions, an administrator can set these limits at a user or group level. The following status categories are available in this badge:

Badge Description
The number of environments is less than 75% of the environment limit.
The number of environments is equal to or greater than 75% of the environment limit.
The number of environments is equal to the environment limit. You can't add more environments without removing an environment first.

Navigate to Model Registry > Custom Model Workshop and select the Environments tab. This tab lists the environments provided by DataRobot and those you have created. Click Add new environment to configure the environment details and add it to the workshop.

Complete the fields in the Add New Environment dialog box.

Field Description
Environment name The name of the environment.
Choose the file you want to upload The tarball archive containing the Dockerfile and any other relevant files.
Programming Language The language in which the environment was made.
Description (optional) An optional description of the custom environment.

When all fields are complete, click Add. The custom environment is ready for use in the Workshop.

Add an environment version

Troubleshoot or update a custom environment by adding a new version of it to the Workshop. In the Versions tab, select New version.

Upload the file for a new version and provide a brief description, then click Add.

The new version is available in the Verison tab; all past environment versions are saved for later use.


Updated November 22, 2022
Back to top