Skip to content

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

DataRobot Prime examples

You can generate source code for the model as a Python module or Java class.

Using DataRobot Prime with Python

DataRobot Prime with Python requires:

  • Python (Recommended: 3.7)
  • Numpy (Recommended: 1.16)
  • Pandas < 1.0 (Recommended: 0.23)

To make predictions with a DataRobot Prime model, run the exported Python script file using the following command:

    python <prediction_file> --encoding=<encoding> <data_file> <output_file>

Where:

  • <prediction_file> specifies the downloaded Python code version of the DataRobot Prime model.
  • <encoding> (optional) specifies the encoding of the dataset you are going to make predictions with. DataRobot Prime defaults to UTF-8 if not otherwise specified. See the "Codecs" column of the Python-supported standards chart for possible alternative entries.
  • <data_file> specifies a .csv file (your dataset); columns must correspond to the feature set used to generate the DataRobot Prime model.
  • <output_file> specifies the filename where DataRobot writes the results.

Python Example

In the following example, prime.py is a Python script containing a DataRobot Prime model trained on the following dataset:

race,gender,age,readmitted
Caucasian,Female,[50-60),0
Caucasian,Male,[50-60),0
Caucasian,Female,[80-90),1

The following command produces predictions for the data in data.csv and outputs the results to results.csv.

python prime.py data.csv results.csv

The file data.csv is a .csv file that looks like this:

race,gender,age
Hispanic,Male,[40-50)
Caucasian,Male,[80-90)
AfricanAmerican,Male,[60-70)

The results in results.csv look like this:

Index,Prediction
0,0.438665626555
1,0.611403738867
2,0.269324648106

Using DataRobot Prime with Java

To run DataRobot Prime with Java:

  • You must use the JDK for Java version 1.7.x or later.
  • Do not rename any of the classes in the file.
  • You must include the Apache commons CSV library version 1.1 or later to be able to run the code.
  • You must rename the exported Prime scoring code Java file to Prediction.java.

Compile the Java file using the following command:

javac -cp ./:./commons-csv-1.1.jar Prediction.java -d ./ -encoding 'UTF-8'

Execute the compiled Java class using the following command:

 java -cp ./:./commons-csv-1.1.jar Prediction <data file> <output file>

Where:

  • <data_file> specifies a .csv file (your dataset); columns must correspond to the feature set used to generate the DataRobot Prime model.
  • <output_file> specifies the filename where DataRobot writes the results.

Java Example

The following example generates predictions for data.csv and writes them to results.csv:

javac -cp ./:./commons-csv-1.1.jar Prediction.java -d ./ -encoding 'UTF-8'
java -cp ./:./commons-csv-1.1.jar Prediction data.csv results.csv

See the Python example for details on the format of input / output data.


Updated October 27, 2021
Back to top