Prediction API snippets¶
DataRobot provides sample Python code containing the commands and identifiers required to submit a CSV or JSON file for scoring. You can use this code with the DataRobot Prediction API. To use the Prediction API Scripting Code, open the deployment you want to make predictions through and click Predictions > Prediction API. On the Prediction API Scripting Code page, you can choose from several scripts for Batch and Real-time predictions. Follow the sample provided and make the necessary changes when you want to integrate the model, via API, into your production application.
Dormant prediction servers
Prediction servers become dormant after a prolonged period of inactivity. If you see the Prediction server is dormant alert, contact support@datarobot.com for reactivation.
Batch prediction snippet settings¶
To find and access the batch prediction script required for your use case, configure the following settings:
Content | Description | |
---|---|---|
1 | Prediction type | Determines the prediction method used. Select Batch. |
2 | Interface | Determines the interface type of the batch prediction script you generate. Select one of the following interfaces:
|
3 | Platform (for CLI interface only) |
Determines the OS on which you intend to run the generated CLI prediction script when you select the CLI interface option. Select one of the following platform types:
|
4 | Show secrets | Displays any secrets hidden by ***** in the code snippet. Revealing the secrets in a code snippet can provide a convenient way to retrieve your API key or datarobot-key ; however, these secrets are hidden by default for security reasons, so ensure that you handle them carefully. |
5 | Copy script to clipboard | Copies the entire code snippet to your clipboard. |
6 | Open in a codespace | Open the snippet in a codespace to edit it, share with others, and incorporate additional files. |
7 | Code overview screen | Displays the example code you can download and run on your local machine. Edit this code snippet to fit your needs. |
Real-time prediction snippet settings¶
To find and access the real-time prediction script required for your use case, configure the following settings:
Content | Description | |
---|---|---|
1 | Prediction type | Determines the prediction method used. Select Real time. |
2 | Language | Determines the language of the real-time prediction script generated. Select a format:
|
3 | Show secrets | Displays any secrets hidden by ***** in the code snippet. Revealing the secrets in a code snippet can provide a convenient way to retrieve your API key or datarobot-key ; however, these secrets are hidden by default for security reasons, so ensure that you handle them carefully. |
4 | Copy script to clipboard | Copies the entire code snippet to your clipboard. |
5 | Open in a codespace | Open the snippet in a codespace to edit it, share with others, and incorporate additional files. |
6 | Code overview screen | Displays the example code you can download and run on your local machine. Edit this code snippet to fit your needs. |
Disable data drift¶
You can disable data drift tracking for individual prediction requests by applying a unique header to the request. This may be useful, for example, in the case where you are using synthetic data that does not have real-world consequences.
Insert the header, X-DataRobot-Skip-Drift-Tracking=1
, into the request snippet. For example:
headers['X-DataRobot-Skip-Drift-Tracking'] = '1'
requests.post(url, auth=(USERNAME, API_KEY), data=data, headers=headers)
After you apply this header, drift tracking is not calculated for the request. However, service stats are still provided (data errors, system errors, execution time, and more).
Open snippets in a codespace¶
You can open a Prediction API code snippet in a codespace to edit the snippet directly, share it with other users, and incorproate additional files.
To open a Prediction API snippet, click Open in a codespace.
DataRobot generates a codespace instance and populates the snippet inside as a python file.
In the codespace, you can upload files and edit the snippet as needed. For example, you may want to add CLI arguments in order to execute the snippet.
The codespace allows for full access to file storage. You can use the Upload button to add additional datasets for scoring, and have the prediction output (output.json
, output.csv
, etc.) return to the codespace file directory after executing the snippet. This example uploads 10k_diabetes_small.csv
to the codespace as an input file.
To add CLI arguments to the snippet, click Add CLI arguments .
This example references 10k_diabetes_small.csv
as the input file for scoring, and names the output file output.csv
.
The snippet is now configured to run and return predictions. When you have finished working in the codespace, click Exit and save codespace.
Codespaces belong to Use Cases, so you must specify an existing Use Case or create a new one to save the codespace to. When a Use Case has been selected, click Exit and save codespace again. Your snippet is now saved in a codespace as part of a Use Case.