Scikit-Learn runtime for MLServer
Project description
Scikit-Learn runtime for MLServer
This package provides a MLServer runtime compatible with Scikit-Learn.
Usage
You can install the runtime, alongside mlserver
, as:
pip install mlserver mlserver-sklearn
For further information on how to use MLServer with Scikit-Learn, you can check out this worked out example.
Content Types
If no content type is present on the request or metadata, the Scikit-Learn runtime will try to decode the payload as a NumPy Array. To avoid this, either send a different content type explicitly, or define the correct one as part of your model's metadata.
Model Outputs
The Scikit-Learn inference runtime exposes a number of outputs depending on the
model type.
These outputs match to the predict
, predict_proba
and transform
methods
of the Scikit-Learn model.
Output | Returned By Default | Availability |
---|---|---|
predict |
✅ | Available on most models, but not in Scikit-Learn pipelines. |
predict_proba |
❌ | Only available on non-regressor models. |
transform |
❌ | Only available on Scikit-Learn pipelines. |
By default, the runtime will only return the output of predict
.
However, you are able to control which outputs you want back through the
outputs
field of your {class}InferenceRequest <mlserver.types.InferenceRequest>
payload.
For example, to only return the model's predict_proba
output, you could
define a payload such as:
---
emphasize-lines: 10-12
---
{
"inputs": [
{
"name": "my-input",
"datatype": "INT32",
"shape": [2, 2],
"data": [1, 2, 3, 4]
}
],
"outputs": [
{ "name": "predict_proba" }
]
}
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
Built Distribution
Hashes for mlserver_sklearn-1.5.0rc1.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2c057357b04203075519831b61978ca7d8642e8c66c0740063d97a4b642dfa9e |
|
MD5 | 4f7b36ebe77cb6b09c92cd230013c06c |
|
BLAKE2b-256 | 0d8429f778bd466d3df103f35c1ff3c2f0f3f3bda7e1414bdd12e34432c8d402 |
Hashes for mlserver_sklearn-1.5.0rc1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1154be04e53cff9af1371f9fef468149760f305f65868bacca6738ce81fa0bf3 |
|
MD5 | 2cefb846b35c355569ea299828838013 |
|
BLAKE2b-256 | af7d9a8a67e97b4c3d0f7f34e25f44342fa7a8415a1694f0dd0cb1a26e501007 |