Skip to content

How to professionally package your ML model for production.

Notifications You must be signed in to change notification settings

simplesoftMX/ML-Model-API-AWS-Fargate

 
 

Repository files navigation

Run With tox (Recommended):

  • Install tox using: pip install tox
  • Make sure you are in the classification_model directory (where the tox.ini file is) then run the command: tox (this runs the tests and typechecks, trains the model under the hood). The first time you run this it creates a virtual env and installs dependencies, so takes a few minutes.

Run Without tox:

  • Add ML-Model-Packaging and classification_model paths to your system PYTHONPATH
  • pip install -r requirements/test_requirements
  • Train the model: python classification_model/train_pipeline.py
  • Run the tests pytest tests

About

How to professionally package your ML model for production.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 62.4%
  • Jupyter Notebook 35.3%
  • Dockerfile 2.1%
  • Other 0.2%