Braindecode 1.2.0.dev184328194__tar.gz → 1.3.0.dev167809764__tar.gz

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 (158) hide show
  1. {braindecode-1.2.0.dev184328194/braindecode.egg-info → braindecode-1.3.0.dev167809764}/PKG-INFO +47 -13
  2. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/README.rst +14 -1
  3. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/augmentation/__init__.py +4 -0
  4. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/augmentation/base.py +1 -1
  5. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/augmentation/functional.py +255 -54
  6. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/augmentation/transforms.py +76 -2
  7. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/classifier.py +3 -8
  8. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datasets/__init__.py +16 -4
  9. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datasets/base.py +132 -153
  10. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datasets/bcicomp.py +4 -4
  11. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datasets/bids.py +3 -3
  12. braindecode-1.3.0.dev167809764/braindecode/datasets/chb_mit.py +172 -0
  13. braindecode-1.3.0.dev167809764/braindecode/datasets/experimental.py +218 -0
  14. braindecode-1.3.0.dev167809764/braindecode/datasets/hub.py +945 -0
  15. braindecode-1.3.0.dev167809764/braindecode/datasets/hub_validation.py +113 -0
  16. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datasets/mne.py +3 -5
  17. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datasets/moabb.py +17 -7
  18. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datasets/nmt.py +40 -45
  19. braindecode-1.3.0.dev167809764/braindecode/datasets/registry.py +120 -0
  20. braindecode-1.3.0.dev167809764/braindecode/datasets/siena.py +164 -0
  21. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datasets/sleep_physio_challe_18.py +4 -3
  22. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datasets/sleep_physionet.py +2 -2
  23. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datasets/tuh.py +2 -2
  24. braindecode-1.3.0.dev167809764/braindecode/datasets/utils.py +67 -0
  25. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datasets/xy.py +2 -2
  26. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datautil/__init__.py +13 -1
  27. braindecode-1.3.0.dev167809764/braindecode/datautil/channel_utils.py +114 -0
  28. braindecode-1.3.0.dev167809764/braindecode/datautil/hub_formats.py +180 -0
  29. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datautil/serialization.py +18 -14
  30. braindecode-1.3.0.dev167809764/braindecode/datautil/util.py +156 -0
  31. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/eegneuralnet.py +22 -48
  32. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/functional/functions.py +6 -2
  33. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/functional/initialization.py +2 -3
  34. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/__init__.py +24 -9
  35. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/atcnet.py +166 -27
  36. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/attentionbasenet.py +152 -20
  37. braindecode-1.2.0.dev184328194/braindecode/models/sleep_stager_eldele_2021.py → braindecode-1.3.0.dev167809764/braindecode/models/attn_sleep.py +23 -9
  38. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/base.py +280 -2
  39. braindecode-1.3.0.dev167809764/braindecode/models/bendr.py +469 -0
  40. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/biot.py +28 -24
  41. braindecode-1.3.0.dev167809764/braindecode/models/config.py +230 -0
  42. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/contrawr.py +4 -2
  43. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/ctnet.py +42 -37
  44. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/deep4.py +10 -6
  45. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/deepsleepnet.py +131 -9
  46. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/eegconformer.py +137 -33
  47. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/eeginception_erp.py +83 -8
  48. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/eeginception_mi.py +5 -3
  49. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/eegitnet.py +5 -3
  50. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/eegminer.py +2 -0
  51. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/eegnet.py +65 -178
  52. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/eegnex.py +114 -7
  53. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/eegsimpleconv.py +3 -1
  54. braindecode-1.3.0.dev167809764/braindecode/models/eegsym.py +917 -0
  55. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/eegtcnet.py +6 -4
  56. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/fbcnet.py +6 -2
  57. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/fblightconvnet.py +3 -1
  58. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/fbmsnet.py +21 -7
  59. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/hybrid.py +1 -1
  60. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/ifnet.py +4 -2
  61. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/labram.py +221 -115
  62. braindecode-1.3.0.dev167809764/braindecode/models/luna.py +836 -0
  63. braindecode-1.3.0.dev167809764/braindecode/models/medformer.py +758 -0
  64. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/msvtnet.py +11 -9
  65. braindecode-1.3.0.dev167809764/braindecode/models/patchedtransformer.py +640 -0
  66. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/sccnet.py +82 -9
  67. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/shallow_fbcsp.py +6 -2
  68. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/signal_jepa.py +111 -27
  69. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/sinc_shallow.py +13 -11
  70. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/sleep_stager_blanco_2020.py +3 -1
  71. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/sleep_stager_chambon_2018.py +3 -1
  72. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/sparcnet.py +6 -4
  73. braindecode-1.3.0.dev167809764/braindecode/models/sstdpn.py +869 -0
  74. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/summary.csv +45 -41
  75. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/syncnet.py +3 -1
  76. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/tcn.py +5 -3
  77. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/tidnet.py +3 -1
  78. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/tsinception.py +16 -4
  79. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/models/usleep.py +111 -12
  80. braindecode-1.3.0.dev167809764/braindecode/models/util.py +358 -0
  81. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/modules/attention.py +10 -10
  82. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/modules/blocks.py +6 -4
  83. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/modules/filter.py +2 -9
  84. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/modules/layers.py +18 -17
  85. braindecode-1.3.0.dev167809764/braindecode/preprocessing/__init__.py +271 -0
  86. braindecode-1.3.0.dev167809764/braindecode/preprocessing/eegprep_preprocess.py +1317 -0
  87. braindecode-1.3.0.dev167809764/braindecode/preprocessing/mne_preprocess.py +240 -0
  88. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/preprocessing/preprocess.py +176 -78
  89. braindecode-1.3.0.dev167809764/braindecode/preprocessing/util.py +177 -0
  90. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/preprocessing/windowers.py +26 -20
  91. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/regressor.py +3 -7
  92. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/samplers/base.py +8 -8
  93. braindecode-1.3.0.dev167809764/braindecode/version.py +1 -0
  94. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764/braindecode.egg-info}/PKG-INFO +47 -13
  95. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode.egg-info/SOURCES.txt +33 -4
  96. braindecode-1.3.0.dev167809764/braindecode.egg-info/requires.txt +67 -0
  97. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/api.rst +152 -17
  98. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/cite.rst +9 -2
  99. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/conf.py +101 -26
  100. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/index.rst +4 -4
  101. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/install/install.rst +1 -1
  102. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/install/install_pip.rst +7 -1
  103. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/install/install_source.rst +1 -1
  104. braindecode-1.3.0.dev167809764/docs/models/categorization/attention.rst +31 -0
  105. braindecode-1.3.0.dev167809764/docs/models/categorization/channel.rst +31 -0
  106. braindecode-1.3.0.dev167809764/docs/models/categorization/convolution.rst +33 -0
  107. braindecode-1.3.0.dev167809764/docs/models/categorization/filterbank.rst +31 -0
  108. braindecode-1.3.0.dev167809764/docs/models/categorization/gnn.rst +27 -0
  109. braindecode-1.3.0.dev167809764/docs/models/categorization/interpretable.rst +31 -0
  110. braindecode-1.3.0.dev167809764/docs/models/categorization/lbm.rst +31 -0
  111. braindecode-1.3.0.dev167809764/docs/models/categorization/recurrent.rst +31 -0
  112. braindecode-1.3.0.dev167809764/docs/models/categorization/spd.rst +31 -0
  113. braindecode-1.3.0.dev167809764/docs/models/models.rst +78 -0
  114. braindecode-1.3.0.dev167809764/docs/models/models_categorization.rst +201 -0
  115. braindecode-1.3.0.dev167809764/docs/models/models_table.rst +191 -0
  116. braindecode-1.3.0.dev167809764/docs/models/models_visualization.rst +31 -0
  117. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/sg_execution_times.rst +65 -62
  118. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/whats_new.rst +81 -2
  119. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/pyproject.toml +41 -15
  120. braindecode-1.2.0.dev184328194/braindecode/datautil/util.py +0 -41
  121. braindecode-1.2.0.dev184328194/braindecode/models/eegresnet.py +0 -362
  122. braindecode-1.2.0.dev184328194/braindecode/models/util.py +0 -129
  123. braindecode-1.2.0.dev184328194/braindecode/preprocessing/__init__.py +0 -37
  124. braindecode-1.2.0.dev184328194/braindecode/preprocessing/mne_preprocess.py +0 -77
  125. braindecode-1.2.0.dev184328194/braindecode/version.py +0 -1
  126. braindecode-1.2.0.dev184328194/braindecode.egg-info/requires.txt +0 -44
  127. braindecode-1.2.0.dev184328194/docs/models_summary.rst +0 -134
  128. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/LICENSE.txt +0 -0
  129. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/MANIFEST.in +0 -0
  130. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/NOTICE.txt +0 -0
  131. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/__init__.py +0 -0
  132. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/datasets/bbci.py +0 -0
  133. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/functional/__init__.py +0 -0
  134. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/modules/__init__.py +0 -0
  135. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/modules/activation.py +0 -0
  136. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/modules/convolution.py +0 -0
  137. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/modules/linear.py +0 -0
  138. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/modules/parametrization.py +0 -0
  139. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/modules/stats.py +0 -0
  140. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/modules/util.py +0 -0
  141. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/modules/wrapper.py +0 -0
  142. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/samplers/__init__.py +0 -0
  143. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/samplers/ssl.py +0 -0
  144. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/training/__init__.py +0 -0
  145. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/training/callbacks.py +0 -0
  146. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/training/losses.py +0 -0
  147. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/training/scoring.py +0 -0
  148. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/util.py +0 -0
  149. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/visualization/__init__.py +0 -0
  150. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/visualization/confusion_matrices.py +0 -0
  151. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode/visualization/gradients.py +0 -0
  152. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode.egg-info/dependency_links.txt +0 -0
  153. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/braindecode.egg-info/top_level.txt +0 -0
  154. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/Makefile +0 -0
  155. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/_templates/autosummary/class.rst +0 -0
  156. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/_templates/autosummary/function.rst +0 -0
  157. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/docs/help.rst +0 -0
  158. {braindecode-1.2.0.dev184328194 → braindecode-1.3.0.dev167809764}/setup.cfg +0 -0
@@ -1,8 +1,8 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: braindecode
3
- Version: 1.2.0.dev184328194
3
+ Version: 1.3.0.dev167809764
4
4
  Summary: Deep learning software to decode EEG, ECG or MEG signals
5
- Author-email: Robin Tibor Schirrmeister <robintibor@gmail.com>
5
+ Author-email: Robin Tibor Schirrmeister <robintibor@gmail.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Alexandre Gramfort <agramfort@meta.com>
6
6
  Maintainer-email: Alexandre Gramfort <agramfort@meta.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Robin Tibor Schirrmeister <robintibor@gmail.com>
7
7
  License: BSD-3-Clause
8
8
  Project-URL: homepage, https://braindecode.org
@@ -14,41 +14,54 @@ Classifier: Intended Audience :: Developers
14
14
  Classifier: Intended Audience :: Science/Research
15
15
  Classifier: Topic :: Software Development :: Build Tools
16
16
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
17
- Classifier: Programming Language :: Python :: 3.10
18
17
  Classifier: Programming Language :: Python :: 3.11
19
18
  Classifier: Programming Language :: Python :: 3.12
20
- Requires-Python: >3.10
19
+ Classifier: Programming Language :: Python :: 3.13
20
+ Classifier: Programming Language :: Python :: 3.14
21
+ Requires-Python: >=3.11
21
22
  Description-Content-Type: text/x-rst
22
23
  License-File: LICENSE.txt
23
24
  License-File: NOTICE.txt
24
- Requires-Dist: mne>=1.10.0
25
+ Requires-Dist: mne>=1.11.0
25
26
  Requires-Dist: mne_bids>=0.16
26
27
  Requires-Dist: numpy
27
28
  Requires-Dist: pandas
28
29
  Requires-Dist: scipy
29
30
  Requires-Dist: matplotlib
30
31
  Requires-Dist: h5py
31
- Requires-Dist: skorch>=1.2.0
32
- Requires-Dist: torch<3.0,>=2.0
33
- Requires-Dist: torchaudio<3.0,>=2.0
32
+ Requires-Dist: skorch>=1.3.0
33
+ Requires-Dist: torch>=2.0
34
+ Requires-Dist: torchaudio>=2.0
34
35
  Requires-Dist: einops
35
36
  Requires-Dist: joblib
36
- Requires-Dist: torchinfo~=1.8
37
+ Requires-Dist: torchinfo
37
38
  Requires-Dist: wfdb
38
- Requires-Dist: h5py
39
39
  Requires-Dist: linear_attention_transformer
40
40
  Requires-Dist: docstring_inheritance
41
+ Requires-Dist: rotary_embedding_torch
41
42
  Provides-Extra: moabb
42
- Requires-Dist: moabb>=1.2.0; extra == "moabb"
43
+ Requires-Dist: moabb>=1.4.2; extra == "moabb"
44
+ Provides-Extra: eegprep
45
+ Requires-Dist: eegprep[eeglabio]>=0.2.23; extra == "eegprep"
46
+ Provides-Extra: hub
47
+ Requires-Dist: huggingface_hub[torch]>=0.20.0; extra == "hub"
48
+ Requires-Dist: zarr>=3.0; extra == "hub"
43
49
  Provides-Extra: tests
44
50
  Requires-Dist: pytest; extra == "tests"
45
51
  Requires-Dist: pytest-cov; extra == "tests"
46
52
  Requires-Dist: codecov; extra == "tests"
47
53
  Requires-Dist: pytest_cases; extra == "tests"
48
54
  Requires-Dist: mypy; extra == "tests"
55
+ Provides-Extra: typing
56
+ Requires-Dist: pydantic<3.0,>=2.0; extra == "typing"
57
+ Requires-Dist: numpydantic>=1.7; extra == "typing"
49
58
  Provides-Extra: docs
50
59
  Requires-Dist: sphinx_gallery; extra == "docs"
51
60
  Requires-Dist: sphinx_rtd_theme; extra == "docs"
61
+ Requires-Dist: sphinx-autodoc-typehints; extra == "docs"
62
+ Requires-Dist: sphinx-autobuild; extra == "docs"
63
+ Requires-Dist: sphinxcontrib-bibtex; extra == "docs"
64
+ Requires-Dist: sphinx_sitemap; extra == "docs"
52
65
  Requires-Dist: pydata_sphinx_theme; extra == "docs"
53
66
  Requires-Dist: numpydoc; extra == "docs"
54
67
  Requires-Dist: memory_profiler; extra == "docs"
@@ -59,10 +72,22 @@ Requires-Dist: lightning; extra == "docs"
59
72
  Requires-Dist: seaborn; extra == "docs"
60
73
  Requires-Dist: pre-commit; extra == "docs"
61
74
  Requires-Dist: openneuro-py; extra == "docs"
75
+ Requires-Dist: plotly; extra == "docs"
76
+ Requires-Dist: shap; extra == "docs"
77
+ Requires-Dist: nbformat; extra == "docs"
62
78
  Provides-Extra: all
63
- Requires-Dist: braindecode[docs,moabb,tests]; extra == "all"
79
+ Requires-Dist: braindecode[moabb]; extra == "all"
80
+ Requires-Dist: braindecode[tests]; extra == "all"
81
+ Requires-Dist: braindecode[docs]; extra == "all"
82
+ Requires-Dist: braindecode[hub]; extra == "all"
83
+ Requires-Dist: braindecode[eegprep]; extra == "all"
84
+ Requires-Dist: braindecode[typing]; extra == "all"
64
85
  Dynamic: license-file
65
86
 
87
+ .. image:: https://zenodo.org/badge/232335424.svg
88
+ :target: https://doi.org/10.5281/zenodo.8214376
89
+ :alt: DOI
90
+
66
91
  .. image:: https://badges.gitter.im/braindecodechat/community.svg
67
92
  :alt: Join the chat at https://gitter.im/braindecodechat/community
68
93
  :target: https://gitter.im/braindecodechat/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge
@@ -78,6 +103,7 @@ Dynamic: license-file
78
103
  :target: https://codecov.io/gh/braindecode/braindecode
79
104
  :alt: Code Coverage
80
105
 
106
+
81
107
  .. |Braindecode| image:: https://user-images.githubusercontent.com/42702466/177958779-b00628aa-9155-4c51-96d1-d8c345aff575.svg
82
108
  .. _braindecode: braindecode.org/
83
109
 
@@ -134,7 +160,15 @@ https://gitter.im/braindecodechat/community
134
160
  Citing
135
161
  ======
136
162
 
137
- If you use this code in a scientific publication, please cite us as:
163
+ If you use Braindecode in scientific work, please cite the software using the Zenodo DOI shown in the badge below:
164
+
165
+ .. image:: https://zenodo.org/badge/232335424.svg
166
+ :target: https://doi.org/10.5281/zenodo.8214376
167
+ :alt: DOI
168
+
169
+ Additionally, we highly encourage you to cite the article that originally introduced
170
+ the Braindecode library and has served as a foundational reference for many works
171
+ on deep learning with EEG recordings. Please use the following reference:
138
172
 
139
173
  .. code-block:: bibtex
140
174
 
@@ -1,3 +1,7 @@
1
+ .. image:: https://zenodo.org/badge/232335424.svg
2
+ :target: https://doi.org/10.5281/zenodo.8214376
3
+ :alt: DOI
4
+
1
5
  .. image:: https://badges.gitter.im/braindecodechat/community.svg
2
6
  :alt: Join the chat at https://gitter.im/braindecodechat/community
3
7
  :target: https://gitter.im/braindecodechat/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge
@@ -13,6 +17,7 @@
13
17
  :target: https://codecov.io/gh/braindecode/braindecode
14
18
  :alt: Code Coverage
15
19
 
20
+
16
21
  .. |Braindecode| image:: https://user-images.githubusercontent.com/42702466/177958779-b00628aa-9155-4c51-96d1-d8c345aff575.svg
17
22
  .. _braindecode: braindecode.org/
18
23
 
@@ -69,7 +74,15 @@ https://gitter.im/braindecodechat/community
69
74
  Citing
70
75
  ======
71
76
 
72
- If you use this code in a scientific publication, please cite us as:
77
+ If you use Braindecode in scientific work, please cite the software using the Zenodo DOI shown in the badge below:
78
+
79
+ .. image:: https://zenodo.org/badge/232335424.svg
80
+ :target: https://doi.org/10.5281/zenodo.8214376
81
+ :alt: DOI
82
+
83
+ Additionally, we highly encourage you to cite the article that originally introduced
84
+ the Braindecode library and has served as a foundational reference for many works
85
+ on deep learning with EEG recordings. Please use the following reference:
73
86
 
74
87
  .. code-block:: bibtex
75
88
 
@@ -5,8 +5,10 @@ Utilities for data augmentation.
5
5
  from . import functional
6
6
  from .base import AugmentedDataLoader, Compose, IdentityTransform, Transform
7
7
  from .transforms import (
8
+ AmplitudeScale,
8
9
  BandstopFilter,
9
10
  ChannelsDropout,
11
+ ChannelsReref,
10
12
  ChannelsShuffle,
11
13
  ChannelsSymmetry,
12
14
  FrequencyShift,
@@ -46,5 +48,7 @@ __all__ = [
46
48
  "Mixup",
47
49
  "SegmentationReconstruction",
48
50
  "MaskEncoding",
51
+ "AmplitudeScale",
52
+ "ChannelsReref",
49
53
  "functional",
50
54
  ]
@@ -189,7 +189,7 @@ class AugmentedDataLoader(DataLoader):
189
189
 
190
190
  Parameters
191
191
  ----------
192
- dataset : BaseDataset
192
+ dataset : RecordDataset
193
193
  The dataset containing the signals.
194
194
  transforms : list | Transform, optional
195
195
  Transform or sequence of Transform to be applied to each batch.