Import Model

External models can be imported into the Data Science Lab and experimented inside the Notebooks.

Please Note:

  • The External models can be registered to the Data Pipeline module and they can be inferred using the Data Science Lab script runner.

  • Only the Native prediction functionality will work for the External models.

Importing the Model

  • Navigate to the Model tab.

  • Click the Import Model option.

  • The user gets redirected to upload the model file.

  • A notification message appears.

  • The imported model gets added to the model list.

Exporting the Model to Data Pipeline

  • The user can start a new Notebook with wrapper function that includes Data, Imported Model, Predict function, and output Dataset with predictions.

  • Register the Imported Model from the model tab given on the Notebook page.

  • The Register Model confirmation dialog box appears.

  • Click the Yes option.

  • A notification message appears, and the model gets registered.

  • Export the script using the Export functionality provided for the Notebooks on the Notebook List page.

  • The Export to Pipeline window appears.

  • Select a specific script from the Notebook.

  • Select the Next option.

  • Click the Export option from the screen that appears.

  • A notification message appears.

Please Note: The imported model gets registered to the Data Pipeline module.

Accessing the Exported Model within the Pipeline User interface

  • Navigate to the Data Pipeline Workflow editor.

  • Drag the DS Lab Script Runner component and configure it.

  • Select the script name from the drop-down option.

  • The exported model along with the script can be accessed inside the Script Runner component.

  • The user can connect the DS Lab Script Runner component to an Input Event.

  • Run the Pipeline.

  • The model predictions can be generated in the Preview tab of the connected Input Event.

Please Note: Only the Exported Models are accessed through the DS Lab Script Runner component, the other models can be accessed through the Model Runner component inside the Data Pipeline.

Try out the Import Model Functionality yourself

Some of the Sample models and related scripts are provided below for the user to try his hands on this functionality. Please download them by a click, and use them in your Notebook by following the above mentioned steps.

Sample files for Sklearn

Sample files for Keras

Sample files for PyTorch

Please Note: The supported extensions for External models - .pkl, .h5, .pth & .pt

Last updated

Change request #64: Data Science Lab Workflow Page merge2