Braindecode 1.3.0.dev179592623__tar.gz → 1.3.0.dev180760430__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 (155) hide show
  1. {braindecode-1.3.0.dev179592623/braindecode.egg-info → braindecode-1.3.0.dev180760430}/PKG-INFO +41 -13
  2. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/README.rst +14 -1
  3. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/augmentation/__init__.py +4 -0
  4. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/augmentation/base.py +1 -1
  5. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/augmentation/functional.py +255 -54
  6. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/augmentation/transforms.py +76 -2
  7. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/classifier.py +3 -8
  8. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/__init__.py +16 -4
  9. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/base.py +132 -153
  10. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/bcicomp.py +4 -4
  11. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/bids.py +3 -3
  12. braindecode-1.3.0.dev180760430/braindecode/datasets/chb_mit.py +172 -0
  13. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/experimental.py +2 -2
  14. braindecode-1.3.0.dev180760430/braindecode/datasets/hub.py +945 -0
  15. braindecode-1.3.0.dev180760430/braindecode/datasets/hub_validation.py +113 -0
  16. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/mne.py +3 -5
  17. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/moabb.py +17 -7
  18. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/nmt.py +40 -45
  19. braindecode-1.3.0.dev180760430/braindecode/datasets/registry.py +120 -0
  20. braindecode-1.3.0.dev180760430/braindecode/datasets/siena.py +164 -0
  21. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/sleep_physio_challe_18.py +4 -3
  22. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/sleep_physionet.py +2 -2
  23. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/tuh.py +2 -2
  24. braindecode-1.3.0.dev180760430/braindecode/datasets/utils.py +67 -0
  25. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/xy.py +2 -2
  26. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datautil/__init__.py +13 -1
  27. braindecode-1.3.0.dev180760430/braindecode/datautil/channel_utils.py +114 -0
  28. braindecode-1.3.0.dev180760430/braindecode/datautil/hub_formats.py +180 -0
  29. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datautil/serialization.py +18 -14
  30. braindecode-1.3.0.dev180760430/braindecode/datautil/util.py +156 -0
  31. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/eegneuralnet.py +22 -48
  32. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/functional/functions.py +6 -2
  33. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/functional/initialization.py +2 -3
  34. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/__init__.py +18 -1
  35. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/atcnet.py +37 -38
  36. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/attentionbasenet.py +43 -36
  37. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/attn_sleep.py +11 -7
  38. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/base.py +280 -2
  39. braindecode-1.3.0.dev180760430/braindecode/models/bendr.py +469 -0
  40. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/biot.py +27 -23
  41. braindecode-1.3.0.dev180760430/braindecode/models/config.py +230 -0
  42. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/contrawr.py +4 -2
  43. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/ctnet.py +41 -36
  44. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/deep4.py +4 -4
  45. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/deepsleepnet.py +32 -23
  46. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/eegconformer.py +29 -24
  47. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/eeginception_erp.py +32 -26
  48. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/eeginception_mi.py +3 -3
  49. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/eegitnet.py +5 -3
  50. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/eegminer.py +2 -0
  51. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/eegnet.py +2 -2
  52. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/eegnex.py +1 -1
  53. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/eegsimpleconv.py +1 -1
  54. braindecode-1.3.0.dev180760430/braindecode/models/eegsym.py +917 -0
  55. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/eegtcnet.py +5 -3
  56. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/fbcnet.py +6 -2
  57. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/fblightconvnet.py +3 -1
  58. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/fbmsnet.py +21 -7
  59. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/hybrid.py +1 -1
  60. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/ifnet.py +4 -2
  61. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/labram.py +221 -115
  62. braindecode-1.3.0.dev180760430/braindecode/models/luna.py +836 -0
  63. braindecode-1.3.0.dev180760430/braindecode/models/medformer.py +758 -0
  64. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/msvtnet.py +11 -9
  65. braindecode-1.3.0.dev180760430/braindecode/models/patchedtransformer.py +640 -0
  66. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/sccnet.py +1 -1
  67. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/shallow_fbcsp.py +4 -2
  68. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/signal_jepa.py +111 -27
  69. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/sinc_shallow.py +13 -11
  70. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/sleep_stager_blanco_2020.py +1 -1
  71. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/sleep_stager_chambon_2018.py +1 -1
  72. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/sparcnet.py +4 -4
  73. braindecode-1.3.0.dev180760430/braindecode/models/sstdpn.py +869 -0
  74. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/summary.csv +7 -1
  75. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/syncnet.py +3 -1
  76. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/tcn.py +5 -3
  77. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/tidnet.py +1 -1
  78. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/tsinception.py +1 -1
  79. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/models/usleep.py +29 -24
  80. braindecode-1.3.0.dev180760430/braindecode/models/util.py +358 -0
  81. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/modules/attention.py +10 -10
  82. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/modules/blocks.py +6 -4
  83. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/modules/filter.py +2 -9
  84. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/modules/layers.py +18 -17
  85. braindecode-1.3.0.dev180760430/braindecode/preprocessing/__init__.py +271 -0
  86. braindecode-1.3.0.dev180760430/braindecode/preprocessing/eegprep_preprocess.py +1317 -0
  87. braindecode-1.3.0.dev180760430/braindecode/preprocessing/mne_preprocess.py +240 -0
  88. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/preprocessing/preprocess.py +157 -53
  89. braindecode-1.3.0.dev180760430/braindecode/preprocessing/util.py +177 -0
  90. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/preprocessing/windowers.py +26 -20
  91. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/regressor.py +3 -7
  92. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/samplers/base.py +8 -8
  93. braindecode-1.3.0.dev180760430/braindecode/version.py +1 -0
  94. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430/braindecode.egg-info}/PKG-INFO +41 -13
  95. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode.egg-info/SOURCES.txt +17 -0
  96. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode.egg-info/requires.txt +25 -8
  97. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/api.rst +143 -13
  98. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/cite.rst +9 -2
  99. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/conf.py +53 -14
  100. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/index.rst +5 -5
  101. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/install/install_pip.rst +7 -1
  102. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/install/install_source.rst +1 -1
  103. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/categorization/attention.rst +2 -2
  104. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/categorization/channel.rst +2 -2
  105. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/categorization/convolution.rst +2 -2
  106. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/categorization/filterbank.rst +3 -3
  107. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/categorization/gnn.rst +3 -6
  108. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/categorization/interpretable.rst +3 -3
  109. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/categorization/lbm.rst +2 -2
  110. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/categorization/recurrent.rst +3 -3
  111. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/categorization/spd.rst +3 -3
  112. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/whats_new.rst +47 -1
  113. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/pyproject.toml +30 -11
  114. braindecode-1.3.0.dev179592623/braindecode/datautil/util.py +0 -41
  115. braindecode-1.3.0.dev179592623/braindecode/models/util.py +0 -129
  116. braindecode-1.3.0.dev179592623/braindecode/preprocessing/__init__.py +0 -37
  117. braindecode-1.3.0.dev179592623/braindecode/preprocessing/mne_preprocess.py +0 -77
  118. braindecode-1.3.0.dev179592623/braindecode/version.py +0 -1
  119. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/LICENSE.txt +0 -0
  120. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/MANIFEST.in +0 -0
  121. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/NOTICE.txt +0 -0
  122. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/__init__.py +0 -0
  123. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/datasets/bbci.py +0 -0
  124. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/functional/__init__.py +0 -0
  125. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/modules/__init__.py +0 -0
  126. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/modules/activation.py +0 -0
  127. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/modules/convolution.py +0 -0
  128. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/modules/linear.py +0 -0
  129. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/modules/parametrization.py +0 -0
  130. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/modules/stats.py +0 -0
  131. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/modules/util.py +0 -0
  132. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/modules/wrapper.py +0 -0
  133. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/samplers/__init__.py +0 -0
  134. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/samplers/ssl.py +0 -0
  135. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/training/__init__.py +0 -0
  136. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/training/callbacks.py +0 -0
  137. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/training/losses.py +0 -0
  138. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/training/scoring.py +0 -0
  139. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/util.py +0 -0
  140. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/visualization/__init__.py +0 -0
  141. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/visualization/confusion_matrices.py +0 -0
  142. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode/visualization/gradients.py +0 -0
  143. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode.egg-info/dependency_links.txt +0 -0
  144. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/braindecode.egg-info/top_level.txt +0 -0
  145. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/Makefile +0 -0
  146. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/_templates/autosummary/class.rst +0 -0
  147. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/_templates/autosummary/function.rst +0 -0
  148. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/help.rst +0 -0
  149. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/install/install.rst +0 -0
  150. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/models.rst +0 -0
  151. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/models_categorization.rst +0 -0
  152. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/models_table.rst +0 -0
  153. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/models/models_visualization.rst +0 -0
  154. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/docs/sg_execution_times.rst +0 -0
  155. {braindecode-1.3.0.dev179592623 → braindecode-1.3.0.dev180760430}/setup.cfg +0 -0
@@ -1,8 +1,8 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: braindecode
3
- Version: 1.3.0.dev179592623
3
+ Version: 1.3.0.dev180760430
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,38 +14,47 @@ 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"
@@ -64,10 +73,20 @@ Requires-Dist: seaborn; extra == "docs"
64
73
  Requires-Dist: pre-commit; extra == "docs"
65
74
  Requires-Dist: openneuro-py; extra == "docs"
66
75
  Requires-Dist: plotly; extra == "docs"
76
+ Requires-Dist: shap; extra == "docs"
67
77
  Provides-Extra: all
68
- Requires-Dist: braindecode[docs,moabb,tests]; extra == "all"
78
+ Requires-Dist: braindecode[moabb]; extra == "all"
79
+ Requires-Dist: braindecode[tests]; extra == "all"
80
+ Requires-Dist: braindecode[docs]; extra == "all"
81
+ Requires-Dist: braindecode[hub]; extra == "all"
82
+ Requires-Dist: braindecode[eegprep]; extra == "all"
83
+ Requires-Dist: braindecode[typing]; extra == "all"
69
84
  Dynamic: license-file
70
85
 
86
+ .. image:: https://zenodo.org/badge/232335424.svg
87
+ :target: https://doi.org/10.5281/zenodo.8214376
88
+ :alt: DOI
89
+
71
90
  .. image:: https://badges.gitter.im/braindecodechat/community.svg
72
91
  :alt: Join the chat at https://gitter.im/braindecodechat/community
73
92
  :target: https://gitter.im/braindecodechat/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge
@@ -83,6 +102,7 @@ Dynamic: license-file
83
102
  :target: https://codecov.io/gh/braindecode/braindecode
84
103
  :alt: Code Coverage
85
104
 
105
+
86
106
  .. |Braindecode| image:: https://user-images.githubusercontent.com/42702466/177958779-b00628aa-9155-4c51-96d1-d8c345aff575.svg
87
107
  .. _braindecode: braindecode.org/
88
108
 
@@ -139,7 +159,15 @@ https://gitter.im/braindecodechat/community
139
159
  Citing
140
160
  ======
141
161
 
142
- If you use this code in a scientific publication, please cite us as:
162
+ If you use Braindecode in scientific work, please cite the software using the Zenodo DOI shown in the badge below:
163
+
164
+ .. image:: https://zenodo.org/badge/232335424.svg
165
+ :target: https://doi.org/10.5281/zenodo.8214376
166
+ :alt: DOI
167
+
168
+ Additionally, we highly encourage you to cite the article that originally introduced
169
+ the Braindecode library and has served as a foundational reference for many works
170
+ on deep learning with EEG recordings. Please use the following reference:
143
171
 
144
172
  .. code-block:: bibtex
145
173
 
@@ -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.