Models

MyoGestic supports following models out of the box: CatBoost, Sklearn, and RaulNet.

All models have the following methods predefined: train, save, load, and predict.

Note

If you wish to add a new model type, you must provide these methods.

CatBoost

train(model, dataset, is_classifier, logger)	Train a CatBoost model.
save(model_path, model)	Save a CatBoost model.
load(model_path, model)	Load a CatBoost model.
predict(model, input, is_classifier)	Predict with a CatBoost model.

Sklearn

train(model, dataset, is_classifier, _)	Train a sklearn model.
save(model_path, model)	Save a sklearn model.
load(model_path, _)	Load a sklearn model.
predict(model, input, is_classifier)	Predict with a sklearn model.

RaulNet

train(model, dataset, _, __)	Train a RaulNet model.
save(_, __)	Return the path of the last saved RaulNet model checkpoint.
load(model_path, model)	Load a RaulNet model from checkpoint.
predict(model, input_data, is_classifier)	Predict with a RaulNet model.