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