---
schema_version: "1"
id: 1b5ff88eeb244c9555d87de7162e07eb4ddf1d62e60871904e93f062e5e0156e
title: Use Databricks Models
collection: admin
version: current
locale: en-us
portal_url: https://docs.salesforce.com/admin/current/en-us/data/264-0-0/c360_a_ai_byom_use_databricks_models.md
canonical_url: https://help.salesforce.com/s/articleView?id=data.c360_a_ai_byom_use_databricks_models.htm&release=264.0.0&type=5
source_site: help.salesforce.com
content_type: documentation
content_hash: 0fe8e040aa759283bac2f4b627174d1a9842dc47831eb961b0f180e3c0e65516
description: You can build, train, and deploy custom machine learning models with Databricks and data in Salesforce. After registering a model in Databricks, define your prediction criteria and connect the model with Data 360 to get predictions and insights for your use case. Use the insights to enrich customer profiles, create segments, and personalize experiences across various channels.
products:
  - Data 360
release: "264"
---
# Use Databricks Models

You can build, train, and deploy custom machine learning models with Databricks and data in Salesforce. After registering a model in Databricks, define your prediction criteria and connect the model with Data 360 to get predictions and insights for your use case. Use the insights to enrich customer profiles, create segments, and personalize experiences across various channels.

### Required Editions

<table class="slds-table slds-table_bordered slds-max-medium-table_stacked slds-m-bottom_small edition"><colgroup><col></colgroup><tbody><tr class=""><td class="slds-cell-wrap" style="vertical-align:top;">Available in: <strong>Developer</strong>, <strong>Enterprise</strong>, <strong>Performance</strong>, and <strong>Unlimited</strong> editions</td></tr></tbody></table>

Here are some benefits of using a Databricks model with Salesforce data.

- Access highly curated, harmonized, and near real-time data across Customer 360. This capability leads to expert models that deliver more intelligent predictions.
- Build models in Databricks and easily connect them with Data 360 to consume predictive insights for business process optimization and drive value. Use cases include propensity to buy, lead conversion, propensity to churn, and more.
- Operationalize models and inferences easily in Salesforce.Data 360 simplifies the consumption of results from business processes and drives value.
- Eliminate tedious, costly, and error-prone Extract, Transform, and Load (ETL) jobs from Databricks using a zero-copy approach to data. This approach reduces costs and improves efficiencies.
- Integrate Vertex AI models and get predictions in Salesforce. Use Flow, a workflow automation suite, to enable AI-driven decision-making that can evolve based on real-time updates to data.

To use a Databricks model with Salesforce:

- Access your Data 360 data in Databricks to build, train, and deploy a model.
- Connect your Databricks model with Data 360 to consume the inferences for predictions and insights in various workflows across Salesforce.

![Diagram showing data flow between Databricks and Salesforce](https://sf-zdocs-cdn-prod.zoominsoftware.com/tdta-data-data_cloud-264-0-0-production-enus/f3569962-3c9b-4f83-9d0c-b6a574e785da/data_cloud/images/c360_a_ai_byom_databricks_data_flow.png)

- **[Set Up Your Data in Databricks](https://help.salesforce.com/s/articleView?id=data.c360_a_ai_byom_set_up_your_data_databricks.htm&release=264.0.0&type=5)**  
    Access Salesforce data in your Databricks notebook using a Python connector. The user-friendly interface lets you quickly execute queries against data in Data 360 so that you can extract the data that you need into Python.
- **[Set Up an Endpoint in Databricks](https://help.salesforce.com/s/articleView?id=data.c360_a_ai_byom_set_up_your_endpoint_databricks.htm&release=264.0.0&type=5)**  
    Control the access to your data by setting up an endpoint connection between Databricks and Data 360.
- **[Connect a Databricks Model](https://help.salesforce.com/s/articleView?id=data.c360_a_ai_byom_connect_your_external_model_databricks.htm&release=264.0.0&type=5)**  
    To consume predictions in Salesforce, connect your Databricks model with Data 360 and define the prediction criteria for your use case.