README.md

Python PyTorch Inference Model Template

This model is intended to work with the Python 3 PyTorch Drop-In Environment. The supplied pth file is a PyTorch 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 PyTorch 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_pytorch --target-type regression --input tests/testdata/juniors_3_year_stats_regression.csv