Skip to content

On-premise users: click in-app to access the full platform documentation for your version of DataRobot.

End-to-end demand forecasting workflow with DataRobot and Databricks

Access this AI accelerator on GitHub

This accelerator is developed for use with Databricks to help you leverage the power of DataRobot for time-series modeling within a Databricks ecosystem.

Demand forecasting models are valuable to many businesses because they apply to high-value use cases such as improving inventory management, supply chain processes, and store staffing. However, building forecasting models can be challenging and time-consuming given the amount of experimentation typically required, from performing time series feature engineering to implementing diverse and complex time-series algorithms and evaluating results. The time series capabilities of DataRobot accelerate this process so you can rapidly build and test many modeling approaches and productionalize your models with model monitoring.

This accelerator can be imported into Databricks notebooks to walk you through how to use DataRobot with Databricks to develop, evaluate, and deploy a multi-series demand forecasting model. The notebook utilizes the DataRobot API to access DataRobot capabilities while ingesting data from Databricks for model building and scoring.

In this accelerator you will:

  • Connect to DataRobot in a Databricks Notebook

  • Import data from Databricks into the AI Catalog

  • Create a time series forecasting project and run Autopilot

  • Retrieve and evaluate model performances and insights

  • Make new predictions with a test dataset

  • Deploy a model with monitoring in DataRobot MLOps

  • Forecast predictions via the Prediction API


Updated October 3, 2023