Skip to content

Instantly share code, notes, and snippets.

@mikesparr

mikesparr/model_serving.py

Created January 5, 2021 06:10
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save mikesparr/6f97d91ebb98a7ca89bd09563e2dd817 to your computer and use it in GitHub Desktop.
Save mikesparr/6f97d91ebb98a7ca89bd09563e2dd817 to your computer and use it in GitHub Desktop.
Download ZIP
AI Demo (model serving)
Raw
model_serving.py
from joblib import dump, load
# save your model (assume you created the function)
dump(my_model, 'model.joblib') # saves local file model.joblib
# load model file later to use for prediction
classifier = load('model.joblib')
# model.predict is a function and features are parameters
prediction = classifier.predict(features) # returns prediction
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment