BaselineRemoval¶
- class torcheeg.transforms.BaselineRemoval[source][source]¶
A transform method to subtract the baseline signal (the signal recorded before the emotional stimulus), the nosie signal is removed from the emotional signal unrelated to the emotional stimulus.
TorchEEG recommends using this class in online_transform for higher processing speed. Even though, this class is also supported in offline_transform. Usually, the baseline needs the same transformation as the experimental signal, please add
apply_to_baseline=True
to all transforms before this operation to ensure that the transformation is performed on the baseline signalfrom torcheeg import transforms t = transforms.Compose([ transforms.BandDifferentialEntropy(apply_to_baseline=True), transforms.ToTensor(apply_to_baseline=True), transforms.BaselineRemoval(), transforms.ToGrid(DEAP_CHANNEL_LOCATION_DICT) ]) t(eeg=np.random.randn(32, 128), baseline=np.random.randn(32, 128))['eeg'].shape >>> (4, 9, 9)