braindecode.training.predict_trials

braindecode.training.predict_trials(module, dataset, return_targets=True)

Create trialwise predictions and optionally also return trialwise labels from cropped dataset given module.

Parameters

module: torch.nn.Module

A pytorch model implementing forward.

dataset: braindecode.datasets.BaseConcatDataset

A braindecode dataset to be predicted.

return_targets: bool

If True, additionally returns the trial targets.

Returns

trial_predictions: np.ndarray

3-dimensional array (n_trials x n_classes x n_predictions), where the number of predictions depend on the chosen window size and the receptive field of the network.

trial_labels: np.ndarray

2-dimensional array (n_trials x n_targets) where the number of targets depends on the decoding paradigm and can be either a single value, multiple values, or a sequence.