braindecode 1.3.0.dev177069446__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (124) hide show
  1. braindecode/__init__.py +9 -0
  2. braindecode/augmentation/__init__.py +52 -0
  3. braindecode/augmentation/base.py +225 -0
  4. braindecode/augmentation/functional.py +1300 -0
  5. braindecode/augmentation/transforms.py +1356 -0
  6. braindecode/classifier.py +258 -0
  7. braindecode/datasets/__init__.py +44 -0
  8. braindecode/datasets/base.py +823 -0
  9. braindecode/datasets/bbci.py +693 -0
  10. braindecode/datasets/bcicomp.py +193 -0
  11. braindecode/datasets/bids/__init__.py +54 -0
  12. braindecode/datasets/bids/datasets.py +239 -0
  13. braindecode/datasets/bids/format.py +717 -0
  14. braindecode/datasets/bids/hub.py +987 -0
  15. braindecode/datasets/bids/hub_format.py +717 -0
  16. braindecode/datasets/bids/hub_io.py +197 -0
  17. braindecode/datasets/bids/hub_validation.py +114 -0
  18. braindecode/datasets/bids/iterable.py +220 -0
  19. braindecode/datasets/chb_mit.py +163 -0
  20. braindecode/datasets/mne.py +170 -0
  21. braindecode/datasets/moabb.py +219 -0
  22. braindecode/datasets/nmt.py +313 -0
  23. braindecode/datasets/registry.py +120 -0
  24. braindecode/datasets/siena.py +162 -0
  25. braindecode/datasets/sleep_physio_challe_18.py +411 -0
  26. braindecode/datasets/sleep_physionet.py +125 -0
  27. braindecode/datasets/tuh.py +591 -0
  28. braindecode/datasets/utils.py +67 -0
  29. braindecode/datasets/xy.py +96 -0
  30. braindecode/datautil/__init__.py +62 -0
  31. braindecode/datautil/channel_utils.py +114 -0
  32. braindecode/datautil/hub_formats.py +180 -0
  33. braindecode/datautil/serialization.py +359 -0
  34. braindecode/datautil/util.py +154 -0
  35. braindecode/eegneuralnet.py +372 -0
  36. braindecode/functional/__init__.py +22 -0
  37. braindecode/functional/functions.py +251 -0
  38. braindecode/functional/initialization.py +47 -0
  39. braindecode/models/__init__.py +117 -0
  40. braindecode/models/atcnet.py +830 -0
  41. braindecode/models/attentionbasenet.py +727 -0
  42. braindecode/models/attn_sleep.py +549 -0
  43. braindecode/models/base.py +574 -0
  44. braindecode/models/bendr.py +493 -0
  45. braindecode/models/biot.py +537 -0
  46. braindecode/models/brainmodule.py +845 -0
  47. braindecode/models/config.py +233 -0
  48. braindecode/models/contrawr.py +319 -0
  49. braindecode/models/ctnet.py +541 -0
  50. braindecode/models/deep4.py +376 -0
  51. braindecode/models/deepsleepnet.py +417 -0
  52. braindecode/models/eegconformer.py +475 -0
  53. braindecode/models/eeginception_erp.py +379 -0
  54. braindecode/models/eeginception_mi.py +379 -0
  55. braindecode/models/eegitnet.py +302 -0
  56. braindecode/models/eegminer.py +256 -0
  57. braindecode/models/eegnet.py +359 -0
  58. braindecode/models/eegnex.py +354 -0
  59. braindecode/models/eegsimpleconv.py +201 -0
  60. braindecode/models/eegsym.py +917 -0
  61. braindecode/models/eegtcnet.py +337 -0
  62. braindecode/models/fbcnet.py +225 -0
  63. braindecode/models/fblightconvnet.py +315 -0
  64. braindecode/models/fbmsnet.py +338 -0
  65. braindecode/models/hybrid.py +126 -0
  66. braindecode/models/ifnet.py +443 -0
  67. braindecode/models/labram.py +1316 -0
  68. braindecode/models/luna.py +891 -0
  69. braindecode/models/medformer.py +760 -0
  70. braindecode/models/msvtnet.py +377 -0
  71. braindecode/models/patchedtransformer.py +640 -0
  72. braindecode/models/reve.py +843 -0
  73. braindecode/models/sccnet.py +280 -0
  74. braindecode/models/shallow_fbcsp.py +212 -0
  75. braindecode/models/signal_jepa.py +1122 -0
  76. braindecode/models/sinc_shallow.py +339 -0
  77. braindecode/models/sleep_stager_blanco_2020.py +169 -0
  78. braindecode/models/sleep_stager_chambon_2018.py +159 -0
  79. braindecode/models/sparcnet.py +426 -0
  80. braindecode/models/sstdpn.py +869 -0
  81. braindecode/models/summary.csv +47 -0
  82. braindecode/models/syncnet.py +234 -0
  83. braindecode/models/tcn.py +275 -0
  84. braindecode/models/tidnet.py +397 -0
  85. braindecode/models/tsinception.py +295 -0
  86. braindecode/models/usleep.py +439 -0
  87. braindecode/models/util.py +369 -0
  88. braindecode/modules/__init__.py +92 -0
  89. braindecode/modules/activation.py +86 -0
  90. braindecode/modules/attention.py +883 -0
  91. braindecode/modules/blocks.py +160 -0
  92. braindecode/modules/convolution.py +330 -0
  93. braindecode/modules/filter.py +654 -0
  94. braindecode/modules/layers.py +216 -0
  95. braindecode/modules/linear.py +70 -0
  96. braindecode/modules/parametrization.py +38 -0
  97. braindecode/modules/stats.py +87 -0
  98. braindecode/modules/util.py +85 -0
  99. braindecode/modules/wrapper.py +90 -0
  100. braindecode/preprocessing/__init__.py +271 -0
  101. braindecode/preprocessing/eegprep_preprocess.py +1317 -0
  102. braindecode/preprocessing/mne_preprocess.py +240 -0
  103. braindecode/preprocessing/preprocess.py +579 -0
  104. braindecode/preprocessing/util.py +177 -0
  105. braindecode/preprocessing/windowers.py +1037 -0
  106. braindecode/regressor.py +234 -0
  107. braindecode/samplers/__init__.py +18 -0
  108. braindecode/samplers/base.py +399 -0
  109. braindecode/samplers/ssl.py +263 -0
  110. braindecode/training/__init__.py +23 -0
  111. braindecode/training/callbacks.py +23 -0
  112. braindecode/training/losses.py +105 -0
  113. braindecode/training/scoring.py +477 -0
  114. braindecode/util.py +419 -0
  115. braindecode/version.py +1 -0
  116. braindecode/visualization/__init__.py +8 -0
  117. braindecode/visualization/confusion_matrices.py +289 -0
  118. braindecode/visualization/gradients.py +62 -0
  119. braindecode-1.3.0.dev177069446.dist-info/METADATA +230 -0
  120. braindecode-1.3.0.dev177069446.dist-info/RECORD +124 -0
  121. braindecode-1.3.0.dev177069446.dist-info/WHEEL +5 -0
  122. braindecode-1.3.0.dev177069446.dist-info/licenses/LICENSE.txt +31 -0
  123. braindecode-1.3.0.dev177069446.dist-info/licenses/NOTICE.txt +20 -0
  124. braindecode-1.3.0.dev177069446.dist-info/top_level.txt +1 -0
braindecode/models/__init__.py ADDED
@@ -0,0 +1,117 @@
1
+ """Some predefined network architectures for EEG decoding."""
2
+
3
+ from .atcnet import ATCNet
4
+ from .attentionbasenet import AttentionBaseNet
5
+ from .attn_sleep import AttnSleep
6
+ from .base import EEGModuleMixin
7
+ from .bendr import BENDR
8
+ from .biot import BIOT
9
+ from .brainmodule import BrainModule
10
+ from .contrawr import ContraWR
11
+ from .ctnet import CTNet
12
+ from .deep4 import Deep4Net
13
+ from .deepsleepnet import DeepSleepNet
14
+ from .eegconformer import EEGConformer
15
+ from .eeginception_erp import EEGInceptionERP
16
+ from .eeginception_mi import EEGInceptionMI
17
+ from .eegitnet import EEGITNet
18
+ from .eegminer import EEGMiner
19
+ from .eegnet import EEGNet, EEGNetv4
20
+ from .eegnex import EEGNeX
21
+ from .eegsimpleconv import EEGSimpleConv
22
+ from .eegsym import EEGSym
23
+ from .eegtcnet import EEGTCNet
24
+ from .fbcnet import FBCNet
25
+ from .fblightconvnet import FBLightConvNet
26
+ from .fbmsnet import FBMSNet
27
+ from .hybrid import HybridNet
28
+ from .ifnet import IFNet
29
+ from .labram import Labram
30
+ from .luna import LUNA
31
+ from .medformer import MEDFormer
32
+ from .msvtnet import MSVTNet
33
+ from .patchedtransformer import PBT
34
+ from .reve import REVE
35
+ from .sccnet import SCCNet
36
+ from .shallow_fbcsp import ShallowFBCSPNet
37
+ from .signal_jepa import (
38
+ SignalJEPA,
39
+ SignalJEPA_Contextual,
40
+ SignalJEPA_PostLocal,
41
+ SignalJEPA_PreLocal,
42
+ )
43
+ from .sinc_shallow import SincShallowNet
44
+ from .sleep_stager_blanco_2020 import SleepStagerBlanco2020
45
+ from .sleep_stager_chambon_2018 import SleepStagerChambon2018
46
+ from .sparcnet import SPARCNet
47
+ from .sstdpn import SSTDPN
48
+ from .syncnet import SyncNet
49
+ from .tcn import BDTCN, TCN
50
+ from .tidnet import TIDNet
51
+ from .tsinception import TSception
52
+ from .usleep import USleep
53
+ from .util import (
54
+ _init_models_dict,
55
+ extract_channel_locations_from_chs_info,
56
+ models_mandatory_parameters,
57
+ )
58
+
59
+ # Call this last in order to make sure the dataset list is populated with
60
+ # the models imported in this file.
61
+ _init_models_dict()
62
+
63
+ __all__ = [
64
+ "ATCNet",
65
+ "AttnSleep",
66
+ "AttentionBaseNet",
67
+ "EEGModuleMixin",
68
+ "BIOT",
69
+ "BENDR",
70
+ "ContraWR",
71
+ "CTNet",
72
+ "Deep4Net",
73
+ "DeepSleepNet",
74
+ "BrainModule",
75
+ "EEGConformer",
76
+ "EEGInceptionERP",
77
+ "EEGInceptionMI",
78
+ "EEGITNet",
79
+ "EEGMiner",
80
+ "EEGNet",
81
+ "EEGNetv4",
82
+ "EEGNeX",
83
+ "EEGSym",
84
+ "EEGSimpleConv",
85
+ "EEGTCNet",
86
+ "FBCNet",
87
+ "FBLightConvNet",
88
+ "FBMSNet",
89
+ "HybridNet",
90
+ "IFNet",
91
+ "Labram",
92
+ "LUNA",
93
+ "extract_channel_locations_from_chs_info",
94
+ "MEDFormer",
95
+ "MSVTNet",
96
+ "PBT",
97
+ "REVE",
98
+ "SCCNet",
99
+ "ShallowFBCSPNet",
100
+ "SignalJEPA",
101
+ "SignalJEPA_Contextual",
102
+ "SignalJEPA_PostLocal",
103
+ "SignalJEPA_PreLocal",
104
+ "SincShallowNet",
105
+ "SSTDPN",
106
+ "SleepStagerBlanco2020",
107
+ "SleepStagerChambon2018",
108
+ "SPARCNet",
109
+ "SyncNet",
110
+ "BDTCN",
111
+ "TCN",
112
+ "TIDNet",
113
+ "TSception",
114
+ "USleep",
115
+ "_init_models_dict",
116
+ "models_mandatory_parameters",
117
+ ]