Exporting models with DataRobot Prime

[#1]

This article showcases how you can use DataRobot Prime to generate rules that approximate models built with DataRobot.

Overview

DataRobot Prime can be initiated on most DataRobot models (see caveats below). It works by creating a series of rules that approximate the original model. Once created, the Prime model can be exported as a Python module or Java class and then be deployed outside of DataRobot.

Creating a DataRobot Prime model

To create a DataRobot Prime model, click on the model of interest and select Predict > DataRobot Prime.

Figure 1. DataRobot Prime tab

In the displayed tab, click Run DataRobot Prime. This will initiate a new model building job; the completed model will appear on the Leaderboard (Figure 2).

Figure 2. DataRobot Prime model

The newly created model includes some functionality that is not included in other, non-Prime DataRobot models. To access that functionality, navigate to the DataRobot Prime tab for the related DataRobot Prime model (Figure 3).

In that tab, you see a graph for rules accuracy. The x-axis indicates the number of rules and the y-axis indicates the accuracy metric currently selected. DataRobot will have optimized the number of rules for the current model based on the accuracy metric but you can manually choose how many rules the final model should have and then kick off another modeling job (Figure 4).

Figure 3. DataRobot Prime Graph

Figure 4. Creation of New DataRobot Prime Model

Exporting the model

Once satisfied with the results, you can download the scoring code by clicking Generate and Download Code. The code can then be used in the environment of your preference.

Caveats

Some important limitations to consider when trying to create a DataRobot Prime model:

• A DataRobot Prime model cannot be created when the original model has been trained into the validation data.
• The feature list of the original model must not contain any user transformations other than var type transformations.
• Multiclass projects are not currently supported.
• Single text feature datasets are not supported.

More information

If you're a licensed DataRobot customer, search in-app documentation for DataRobot Prime tab or DataRobot Prime modeling considerations.