sleepless.engine.predictor_scikit#

Predcition script.

Functions

predict_scikit(dataset, model, ...)

Compute the class probabilities prediction (or prediction if predict probabilities is not possible) for a set of data, given a fitted model.
sleepless.engine.predictor_scikit.predict_scikit(dataset, model, output_folder, model_parameters)[source]#

Compute the class probabilities prediction (or prediction if predict probabilities is not possible) for a set of data, given a fitted model. The prediction are computed for all samples of all keys.

Parameters:

  • dataset (dict[str, list[DelayedSample]]) – A dictionary containing a list of DelayedSample.

  • model (object) – A scikit learn model already fitted.

  • output_folder (str) – A path where prediction will be saved

  • model_parameters (Mapping) – a dictionary where the following key need to be defined, transform: list (if data are not transformed yet)