Skip to content

Latest commit

 

History

History

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 

README.md

Python Sklearn Inference Model Template

This model is intended to work with the Python 3 Scikit-Learn Drop-In Environment. The supplied pkl file is a scikit-learn model trained on juniors_3_year_stats_regression.csv with a Grade 2014 as the target (regression), though any binary or regression model trained using the libraries outlined in Python 3 Scikit-Learn Drop-In Environment will work.

For this sample model, custom.py contains additional data pre-processing that the model itself lacks.

Instructions

Create a new custom model with these files and use the Python Drop-In Environment with it

To run locally using 'drum'

Paths are relative to ./datarobot-user-models:
drum score --code-dir model_templates/python3_sklearn --target-type regression --input tests/testdata/juniors_3_year_stats_regression.csv