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)

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)