Skip to main content

TotoML SDK for Python

Project description

TotoML Python SDK

This library is an SDK for Toto ML

This library provides the following key components:

Model Controller: the Model Controller is a wrapper for your models. It provides:

  • REST APIs to trigger the training, scoring and predictions
  • Event-based access to the same (training, scoring, predicting)

How to use it

In your main python file (e.g. app.py), import the ModelController and ControllerConfig classes, create a Flask app, instantiate your model (in this example ERCBOD()) and pass it to the ModelController constructor.

from totoml.controller import ModelController
from totoml.config import ControllerConfig

app = Flask(__name__)

model_controller = ModelController(ERCBOD(), app, ControllerConfig(enable_batch_predictions_events=False, enable_single_prediction_events=False))

In order for the Model Controller to work, you have to provide to it your model (in the example ERBOCD()): that's called here a Model Delegate.

ModelDelegate is an abstract class provided by this SDK (delegate.ModelDelegate). Your model has to implement its methods.
Those methods are:

get_name()
This method has to return the name of the model. This is very important because the name of the model is used for everything in TotoML (internal folders, registry, file storage, etc..), so the name has to be unique.

get_model_type()
Not very used for now. Should return the type of model: sklearn, tf, etc..
Has to be an instance of ModelType

predict() and predict_batch()
These methods perform predictions.

score()
This method scores the current champion model: it has to recalculate the metrics for the model and return them

train()
This method performs the training of the model and returns the trained model and all the associated files.
It's important to note that also any file containing training data or built features should be returned, so that Toto ML can grant the persistence of those files and make sure there is lineage.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

totoml-1.0.3-py3-none-any.whl (13.8 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page