model_server package#

Submodules#

model_server.model_server module#

class model_server.model_server.ClassifierModel(name: str, model_dir: str | None = None, model=None)[source]#

Bases: MLModelServer

load()[source]#

Load model from storage.

predict(body: dict) List[source]#

Generate model predictions from sample.

:param body : A dict of observations, each of which is an 1-dimensional feature vector.

Returns model predictions as a List, one for each row in the body input List.

Module contents#