DataRobot data connections¶
If your database is protected by a network policy that only allows connections from specific IP addresses, have an administrator add all allowed IPs for DataRobot to your network policy. If the problem persists, contact your DataRobot representative.
The DataRobot connectivity platform allows users to integrate with their data stores using either the DataRobot provided connectors or uploading the JDBC driver provided by the data store.
The "self-service" JDBC database connectivity solution is a standardized, platform-independent solution that does not require complicated installation and configuration. Once configured, you can read data from production databases for model building and predictions. Connectivity to your data source allows you to quickly train and retrain models on that data, and avoids the unnecessary step of exporting data from your database to a CSV file for ingest into DataRobot. It allows access to more diverse data, which results in more accurate models.
By default, only users with "Can manage JDBC database drivers" permission can add, update, or remove JDBC drivers. Only users with "Can manage connectors" permission can add connections. See Roles and permissions for details on permissions.
Users with the technical abilities and permissions can configure and establish data connections. Other users in the org can then leverage those connections to solve business problems. The below sections explains this in more detail:
- An overview of the database connectivity workflow.
- Steps for creating new connections.
- Steps for adding data sources.
- Steps for sharing data connections.
Database connectivity workflow¶
DataRobot's database connectivity workflow, described below, has two fundamental components. First, the administrator uploads JDBC drivers and configures database connections for those drivers. Then, users can import data into DataRobot for project creation and predictions, as follows:
From the Data Connections page, create data connection configuration(s).
Once configured, your data sources are available for both ingest from the Start screen and for predictions from the Make Predictions tab.
There are additional opportunities to launch the data source creation dialogs, but these instructions describe the process used in all cases.
Allowed source IP addresses¶
Any connection initiated from DataRobot originates from one of the following IP addresses:
These IP addresses are reserved for DataRobot use only.
Create a new connection¶
To create a new data connection:
From the account menu on the top right, select Data connections.
Click Add new data connection to open the data store selection dialog box. You can also create a new data connection using the AI Catalog by selecting Add to catalog > New Data Connection.
Select the tile for the data store you wish to use.
Self-Managed AI Platform installations
For Self-Managed AI Platform installations, you might not see any data stores listed. In that case, click Add a new driver and add a driver from the list of supported connections.
Complete the fields for the data store. They will vary slightly based on the data store selected.
Field Description Data connection name Provide a unique name for the connection. Version Select the version of the data store to use from the dropdown list. Configuration: Parameters Modify parameters for connections. Configuration: URL Enter the URL to the database/data store to connect to, in the form jdbc:mysql://<HOST>:<PORT>/<NAME>. You can include parameters in the URL if your connection requires them.
Click Add data connection to save the configuration.
The new connection appears in the left-panel list of Data Connections.
Any connection that you create is only available to you unless you share it with others.
Data connection with parameters¶
The parameters provided for modification in the data connection configuration screen are dependent on the selected driver. Available parameters are dependent on the configuration done by the administrator who added the driver.
Many other fields can be found in a searchable expanded field. If a desired field is not listed, you can click Add parameter to include it.
Click the trash can icon () to remove a listed parameter from the connection configuration.
Additional parameters may be required to establish a connection to your database. These parameters are not always pre-defined in DataRobot, in which case, they must be manually added.
For more information on the required parameters, see the documentation for your database.
Test the connection¶
Once your data connection is created, test the connection by clicking Test connection.
In the resulting dialog box, enter or use stored credentials for the database identified in the JDBC URL field or the parameter-based configuration of the data connection creation screen. Click Sign in and when the test passes successfully, click Close to return to the Data Connections page and create your data sources.
Snowflake and Google BigQuery users can set up a data connection using OAuth single sign-on. Once configured, you can read data from production databases to use for model building and predictions.
Modify a connection¶
You can modify the name, JDBC URL, and, if the driver was configured with them, the parameters of an existing data source.
Select the data connection in the left-panel connections list.
In the updated main window, click in the box of the element you want to edit and enter new text.
Click Save changes.
Delete a connection¶
You can delete any data connection that is not being used by an existing data source. If it is being used, you must first delete the dependencies. To delete a data connection:
From the Data Connections tab, select the data connection in the left-panel connections list.
Click the Delete button in the upper right ().
DataRobot prompts for confirmation. Click Delete to remove the data connection. If there are data sources dependent on the data connection, DataRobot returns a notification.
Once all dependent data sources are removed via the API, try again to delete the data connection.
Add data sources¶
Your data sources specify, via SQL query or selected table and schema data, which data to extract from the data connection. It is the extracted data that you will use for modeling and predictions. You can point to entire database tables or use a SQL query to select specific data from the database. Any data source that you create is available only to you.
Once data sources are created, they cannot be modified and can only be deleted via the API.
To add a data source, do one of the following:
From the Start screen, click Data Source and select the connection that holds the data you would like to add. See how to import from an existing data source.
Share data connections¶
Because the user creating a data connection and the end-user may not be the same, or there may be multiple end-users for the data connection, DataRobot provides the ability to set user-level permissions for each entity. You can accomplish scenarios like the following:
- A user wants to set permissions on a selected data entity to control who has consumer-level, editor-level, or owner-level access. Or, the user wants to remove a particular user's access.
- A user that has had a data connection shared with them wants the shared entity to appear under their list of available entities.
When you invite a user, user group, or organization to share a data connection, DataRobot assigns the default role of Editor to each selected target (not all entities allow sharing beyond a specific user). You can change the role from the dropdown menu.
To share data connections:
From the account menu on the top right, select Data Connections, select a data connection, and click Share:
Enter the email address, group name, or organization you are adding and select a role. Check the box to grant sharing permission.
Click Share to add the user, user group, or organization.
Add any number of collaborators and when finished, click Close to dismiss the sharing dialog box.
Depending on your own permissions, you can remove any user or change access as described in the table of roles and permissions.
There must be at least one Owner for each entity; you cannot remove yourself or remove your sharing ability if you are the only collaborating Owner.