Skip to content
Last updated

AutoML Model Sharing

It is possible to enable sharing of an AutoML model. This enables one user to train the model, and then for a different user to utilize the generated model on their data to make predictions.

When a training model is run, the notebook is only visible to the first user who executed the train notebook. Similarly, if a second user runs a prediction notebook using the model generated by the first user, the resulting notebook is only visible to the second user. This ensures that data security restrictions are maintained.

Share an AutoML Model With Another User

Scenario: User Alice wants to share an AutoML model with User Bob.

  1. User Alice sets the AutoML workflow parameter share_model to True.
    AutoML workflow parameter share_model set to True
  2. Alice executes the training workflow and views the resulting notebook to ensure the model was created correctly.
  3. Alice views the UUID for the AutoML model in the stored parameters:
  4. Navigate to Data Workbench > Workflows.
  5. Select a workflow and open the Run History.
    Workflow run history list in Data Workbench
  6. Select the Run History for a successful run.
    Successful workflow run details
  7. Select the Tasks tab.
  8. For a task with an associated notebook, select the ellipsis ( … ) menu and choose Parameters.
    Task parameters menu
  9. The Stored Params shows the shared_model UUID assigned to this AutoML model. Copy the text inside the quotes (highlighted).
    Stored Params showing shared_model UUID
  10. Send this shared model UUID to User Bob using a secure channel.

Use a Shared AutoML Model for a Prediction

  1. User Bob creates a prediction workflow.
  2. Select the AutoML tab for this new workflow.
  3. Select the notebook workflow to edit.
  4. Set the shared_model parameter to the UUID provided by Alice.
    Prediction workflow AutoML tab with shared_model parameter
  5. When using shared_model, remove the model_name parameter.
  6. Select the Project Revisions tab.
    Project Revisions tab
  7. Edit the workflow to remove the model_name parameter.
    Editing workflow removing model_name parameter
  8. Run the workflow. Confirm in the workflow log that the correct model was used.
    Workflow log confirming selected shared model