Skip to content

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

MLOps agent

Availability information

The MLOps agent is a feature exclusive to DataRobot MLOps. Contact your DataRobot representative for information on enabling it.

DataRobot MLOps provides powerful tools for tracking and managing models for prediction. But what if you already have—or need to have—deployments running in your own environment? Could you use these tools to monitor external models that may have intermittent or no connectivity and so may report predictions sporadically?

The MLOps agent allows you to monitor and manage external models, i.e., those running outside of DataRobot MLOps. With this functionality, predictions and information from these models can be reported as part of MLOps deployments. You can use the same powerful model management tools to monitor accuracy, data drift, prediction distribution, latency, and more, regardless of where the model is running. Data provided to DataRobot MLOps provides valuable insight into the performance and health of those externally deployed models.

The MLOps agent provides:

  • The ability to manage, monitor, and get insight from all model deployments in a single system.
  • API and communications constructed to ensure little or no latency when monitoring external models.
  • Support for deployments that are always connected to the network and the MLOps system, as well as partially or never-connected deployments.
  • The MLOps library is available in Python and Java. It can be used to monitor models written natively in those languages or it can be used to report the input and output of a model artifact in any language.
  • Configuration with the Portable Prediction Server.

Reference the following sections for more information about using the MLOps Agent:

Download the MLOps agent tarball

You can download the MLOps agent tarball from two locations: the Developer Tools page or the Predictions > Monitoring tab of a deployment configured to monitor an external model.

Components of the MLOps agent

When you enable the MLOps agent, you have access to the agent installation and MLOps components, all packaged within a single tarball. The image below illustrates the roles of these components in enabling DataRobot MLOps to manage external models.

External models (1) are machine learning models running outside of DataRobot, within your environment. The deployments (running in Python or Java) score data and generate predictions along with other information, such as the number of predictions created and the length of time to create each.

The MLOps library (2), available in Python (v2 and v3) and Java, provides APIs to report prediction data and information from a specified deployment (identified by deployment ID and model ID). Supported library calls for the MLOps client let you specify which data to report to the MLOps service, including prediction time, number of predictions, and other metrics and deployment statistics. The library-provided APIs buffer messages into a configured location (3). In addition to the MLOps library, the tarball includes Python and Java API examples and accompanying datasets to:

  • create a deployment that generates (example) predictions for both regression and classification models.
  • report metrics from deployments using the MLOps library.

and scripts to:

  • start and stop the agent as well as retrieve current agent status.
  • create a remote deployment that uploads a training dataset and returns the deployment ID and model ID for the deployment.

The MLOps agent (4) detects data written to the target buffer location and reports it to the MLOps service (5). If the agent is running as a service, it retrieves the data as soon as it’s available; otherwise, it retrieves prediction data when it is run manually.

If models are running in isolation and disconnected from the network, the MLOps library will not have networked access from the buffer directory. For these deployments, you can manually copy prediction data from the buffer location via USB drive as needed. The agent then accesses that data as configured and reports it the MLOps service.

Additional agent configuration specifies where to read data from and report data to, how frequently to report the data, and so forth. The flexible MLOps agent design ensures it supports a variety of deployment and environment requirements.

Finally, from the deployment inventory you can view your deployments and view and manage prediction statistics and metrics.

How the agent works

This section outlines the basic workflow for using the MLOps agent from different environments.

Using DataRobot MLOps:

  1. Use the Model Registry to create a model package with information about your model's metadata.
  2. Deploy the model package. Create a deployment to display metrics about the running model.
  3. Use the deployment Predictions tab to view a code snippet demonstrating how to instrument your prediction code with the agent to report metrics.

Using a remote deployment environment:

  1. Install the DataRobot MLOps agent.
  2. Use the MLOps library to report metrics from your prediction code as demonstrated by the snippet. The MLOps library buffers the metrics in a spooler, which enables high throughput without slowing down the deployment. The type of spooler you use depends on your environment. Currently, the spooler can be the local filesystem, RabbitMQ, Amazon SQS, Google PubSub, or Kafka.
  3. The MLOps agent forwards the metrics to DataRobot MLOps.
  4. You can view the reported metrics via the DataRobot MLOps Deployment inventory.


Updated October 7, 2021