model_templates

Custom Inference Model Templates

This folder provides templates for building and deploying custom models in DataRobot. Use the templates as an example structure for your own custom models.

DataRobot User Model Runner

The examples in this repository use the DataRobot User Model Runner (DRUM). For more information on how to use and write models with DRUM, reference the readme.

Sample Models

The model_templates folder contains sample models that work with the provided template environments. For more information about each model, reference the readme for every example:

Inference Models

• Scikit-Learn sample model
• Scikit-Learn sample unsupervised anomaly detection model
• PyTorch sample model
• XGBoost sample model
• Keras sample model
• Keras sample model + Joblib artifact
• ONNX sample binary classification model
• ONNX sample multiclass classification model
• ONNX sample regression model
• ONNX sample anomaly detection model
• PyPMML sample model
• R sample model
• Java sample model
• Julia sample models
• Sample model with runtime parameters
• Proxy Model sample for AzureML
• Proxy Model sample for DataRobot