Predict#

In prediction mode, we input data, the trained model with its weights, and output probabilities prediction which are stored in h5py files.

You may issue sleepless predict --help for a help message containing more detailed instructions.

To run predictions, use the sub-command predict to run prediction on an existing dataset:

sleepless predict -vv <model> <dataset> -w <path/to/model.pth>  -o <path/to/output_folder>

Replace <model> and <dataset> by the appropriate configuration files. Replace <path/to/model.pth> to a path leading to the model weights.

Examples#

To predict with the trained Chambon CNN on the sc-edf dataset:

sleepless train -vv sc_filtered chambon -w <path/to/model.pth> -o <path/to/output_folder>