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 Models tab given on the Notebook page.

  • The Register Model dialog box appears to confirm about the model registration.

  • 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 Validate icon to validate the script.

  • A notification message appears to assure the validity of the script.

  • Click the Export to Pipeline option.

  • A notification message appears to assure that the selected Notebook has been exported.

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

  • Refer the Data Science Lab Quick Start Flow page to get an overview of the Data Science Lab module in nutshell.

Last updated