yasa.SleepStaging.predict_proba

SleepStaging.predict_proba(path_to_model='auto')

Return the predicted probability for each sleep stage for each 30-sec epoch of data.

Currently, only classifiers that were trained using a LGBMClassifier are supported.

Parameters:

path_to_modelstr or “auto”

Full path to a trained LGBMClassifier, exported as a joblib file. Can be “auto” to use YASA’s default classifier.

Returns:

probapandas.DataFrame

The predicted probability for each sleep stage for each 30-sec epoch of data.