Machine learning prediction serving
Project description
ServeIt deploys your trained models to a RESTful API for prediction serving. Current features include:
Model prediction serving
Supplementary information endpoint creation
Configurable request and response logging (work in progress)
Installation: Python 2.7 and Python 3.6
PyPi: pip install serveit
source: git clone https://github.com/rtlee9/serveit.git && cd serveit && pip install -e . # WIP
Supported libraries
Scikit-Learn
from sklearn.datasets import load_iris
from sklearn.linear_model import LogisticRegression
from serveit.sklearn_server import SklearnServer
# fit a model on the Iris dataset
data = load_iris()
reg = LogisticRegression()
reg.fit(data.data, data.target)
# deploy model to a SkLearnServer
sklearn_server = SklearnServer(reg, reg.predict)
# add informational endpoints
sklearn_server.create_model_info_endpoint()
sklearn_server.create_info_endpoint('features', data.feature_names)
sklearn_server.create_info_endpoint('target_labels', data.target_names.tolist())
# start API
sklearn_server.serve()
Then try out your new API:
curl -XPOST 'localhost:5000/predictions'\
-H "Content-Type: application/json"\
-d "[[5.6, 2.9, 3.6, 1.3], [4.4, 2.9, 1.4, 0.2], [5.5, 2.4, 3.8, 1.1], [5.0, 3.4, 1.5, 0.2], [5.7, 2.5, 5.0, 2.0]]"
# [1, 0, 1, 0, 2]
curl -XGET 'localhost:5000/info/model'
# {"penalty": "l2", "tol": 0.0001, "C": 1.0, "classes_": [0, 1, 2], "coef_": [[0.4150, 1.4613, -2.2621, -1.0291], ...], ...}
curl -XGET 'localhost:5000/info/features'
# ["sepal length (cm)", "sepal width (cm)", "petal length (cm)", "petal width (cm)"]
curl -XGET 'localhost:5000/info/target_labels'
# ["setosa", "versicolor", "virginica"]
Coming soon:
TensorFlow
Keras
PyTorch
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
ServeIt-0.0.1a2.tar.gz
(6.6 kB
view hashes)
Built Distribution
Close
Hashes for ServeIt-0.0.1a2-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | bfdeb08f2248b9fe1d3eedb6e655f3c44654c11f396c3441d2a96c768427696d |
|
MD5 | 8468f91f2f5f1a59e50e34eac4781faa |
|
BLAKE2b-256 | f044a78dc31bf4dc95cda388cb1c5748b567def664ab34a277fd0c5fbf259196 |