<div align="center">
<br/>
<div align="left">
<br/>
<p align="center">
<a href="https://github.com/neurocode-ai/neurocode">
<img align="center" width=40% src="https://github.com/neurocode-ai/neurocode/blob/main/docs/images/neurocode_logo.webp"></img>
</a>
</p>
</div>
[](https://python-poetry.org/)
[](https://github.com/astral-sh/ruff)
[](https://github.com/psf/black)
[](https://pypi.org/project/neurocode/)
[](https://opensource.org/licenses/MIT)
[](https://codecov.io/gh/neurocode-ai/neurocode)
[](https://github.com/neurocode-ai/neurocode/actions/workflows/ci.yml)
[](https://github.com/neurocode-ai/neurocode/actions/workflows/tests.yml)
</div>
## 🔎 Overview
A minimalistic Python library for EEG/MEG deep learning research, primarely focused on self-supervised learning.
## 📦 Installation
Either clone this repository and perform a local install accordingly
```
git clone https://github.com/neurocode-ai/neurocode.git
cd neurocode
poetry install
```
or install the most recent release from the Python Package Index (PyPI).
```
pip install neurocode
```
## 🚀 Example usage
Below you can see an example adapted for a SSL training workflow using the SimCLR framework.
```python
import torch
from pytorch_metric_learning import losses
from neurocode.datasets import SimulatedDataset, RecordingDataset
from neurocode.samplers import SignalSampler
from neurocode.models import SignalNet
from neurocode.training import SimCLR
from neurocode.datautil import manifold_plot, history_plot
sample_data = SimulatedDataset("sample", seed=7815891891337)
sample_data.read_from_file("MEG/sample/sample_audvis_raw.fif")
# create random extrapolated data from the raw MEG recording,
# you need to provide a location to a forward solution (source space) to use
sample_data.simulate("MEG/sample/sample_audvis-meg-eeg-oct-6-fwd.fif")
dataset = RecordingDataset(sample_data.data(), sample_data.labels(), sfreq=200)
train, valid = dataset.train_valid_split(split=0.75)
samplers = {
'train': SignalSampler(train.data(), train.labels(), train.info(), ...),
'valid': SignalSampler(valid.data(), valid.labels(), valid.info(), ...),
}
device = "cuda" if torch.cuda.is_available() else "cpu"
model = SignalNet(...)
optimizer = torch.optim.Adam(model.parameters(), ...)
scheduler = torch.optim.lr_scheduler.CosineAnnealingLR(optimizer, ...)
criterion = losses.NTXentLoss()
# train the neural network using the self-supervised learning SimCLR framework,
# save or plot the history to see training loss evolution
simclr = SimCLR(model, device, ...)
history = simclr.fit(samplers, save_model=True)
```
## 📋 License
All code is to be held under a general MIT license, please see [LICENSE](https://github.com/neurocode-ai/neurocode/blob/main/LICENSE) for specific information.
Raw data
{
"_id": null,
"home_page": "https://github.com/neurocode-ai/neurocode",
"name": "neurocode",
"maintainer": "",
"docs_url": null,
"requires_python": ">=3.9,<3.13",
"maintainer_email": "",
"keywords": "deep-learning,meg,eeg,self-supervised-learning,simclr",
"author": "Wilhelm \u00c5gren",
"author_email": "wilhelmagren98@gmail.com",
"download_url": "https://files.pythonhosted.org/packages/25/0a/58c4e297150550bd1f59a901a83a4296f761b7180294ec5795b77786cd63/neurocode-1.0.0.tar.gz",
"platform": null,
"description": "<div align=\"center\">\n<br/>\n<div align=\"left\">\n<br/>\n<p align=\"center\">\n<a href=\"https://github.com/neurocode-ai/neurocode\">\n<img align=\"center\" width=40% src=\"https://github.com/neurocode-ai/neurocode/blob/main/docs/images/neurocode_logo.webp\"></img>\n</a>\n</p>\n</div>\n \n[](https://python-poetry.org/)\n[](https://github.com/astral-sh/ruff)\n[](https://github.com/psf/black)\n[](https://pypi.org/project/neurocode/)\n[](https://opensource.org/licenses/MIT)\n[](https://codecov.io/gh/neurocode-ai/neurocode)\n[](https://github.com/neurocode-ai/neurocode/actions/workflows/ci.yml)\n[](https://github.com/neurocode-ai/neurocode/actions/workflows/tests.yml)\n\n</div>\n\n## \ud83d\udd0e Overview\nA minimalistic Python library for EEG/MEG deep learning research, primarely focused on self-supervised learning. \n\n## \ud83d\udce6 Installation\nEither clone this repository and perform a local install accordingly\n```\ngit clone https://github.com/neurocode-ai/neurocode.git\ncd neurocode\npoetry install\n```\nor install the most recent release from the Python Package Index (PyPI).\n```\npip install neurocode\n```\n\n## \ud83d\ude80 Example usage\nBelow you can see an example adapted for a SSL training workflow using the SimCLR framework.\n\n```python\nimport torch\n\nfrom pytorch_metric_learning import losses\nfrom neurocode.datasets import SimulatedDataset, RecordingDataset\nfrom neurocode.samplers import SignalSampler\nfrom neurocode.models import SignalNet\nfrom neurocode.training import SimCLR\nfrom neurocode.datautil import manifold_plot, history_plot\n\nsample_data = SimulatedDataset(\"sample\", seed=7815891891337)\nsample_data.read_from_file(\"MEG/sample/sample_audvis_raw.fif\")\n\n# create random extrapolated data from the raw MEG recording,\n# you need to provide a location to a forward solution (source space) to use\nsample_data.simulate(\"MEG/sample/sample_audvis-meg-eeg-oct-6-fwd.fif\")\n\ndataset = RecordingDataset(sample_data.data(), sample_data.labels(), sfreq=200)\ntrain, valid = dataset.train_valid_split(split=0.75)\n\nsamplers = {\n 'train': SignalSampler(train.data(), train.labels(), train.info(), ...),\n 'valid': SignalSampler(valid.data(), valid.labels(), valid.info(), ...),\n}\n\ndevice = \"cuda\" if torch.cuda.is_available() else \"cpu\"\nmodel = SignalNet(...)\noptimizer = torch.optim.Adam(model.parameters(), ...)\nscheduler = torch.optim.lr_scheduler.CosineAnnealingLR(optimizer, ...)\ncriterion = losses.NTXentLoss()\n\n# train the neural network using the self-supervised learning SimCLR framework,\n# save or plot the history to see training loss evolution\nsimclr = SimCLR(model, device, ...)\nhistory = simclr.fit(samplers, save_model=True)\n\n```\n\n## \ud83d\udccb License\nAll code is to be held under a general MIT license, please see [LICENSE](https://github.com/neurocode-ai/neurocode/blob/main/LICENSE) for specific information.\n",
"bugtrack_url": null,
"license": "MIT",
"summary": "\ud83e\udde0 EEG/MEG self-supervised learning toolbox.",
"version": "1.0.0",
"project_urls": {
"Bug tracker": "https://github.com/neurocode-ai/neurocode/issues",
"Homepage": "https://github.com/neurocode-ai/neurocode",
"Repository": "https://github.com/neurocode-ai/neurocode"
},
"split_keywords": [
"deep-learning",
"meg",
"eeg",
"self-supervised-learning",
"simclr"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "e5b4b03c81b8cc0a3244cfa3b4ddad976c28f1e184e42bc496a57b63ca54a6c6",
"md5": "df9437d5efcf31754f4b17ac47ea0e57",
"sha256": "66d256484e44d8ace1f652c1a83c8532ce48f94f2614a23f046b4d2bcd637aaa"
},
"downloads": -1,
"filename": "neurocode-1.0.0-py3-none-any.whl",
"has_sig": false,
"md5_digest": "df9437d5efcf31754f4b17ac47ea0e57",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.9,<3.13",
"size": 53087,
"upload_time": "2023-11-19T15:49:04",
"upload_time_iso_8601": "2023-11-19T15:49:04.755474Z",
"url": "https://files.pythonhosted.org/packages/e5/b4/b03c81b8cc0a3244cfa3b4ddad976c28f1e184e42bc496a57b63ca54a6c6/neurocode-1.0.0-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "250a58c4e297150550bd1f59a901a83a4296f761b7180294ec5795b77786cd63",
"md5": "24772f71bedbc4d014f0589400c4db72",
"sha256": "e746837720a48169db9ffa3429f1c82108c60c988fa1e140fb9078e97d9b961b"
},
"downloads": -1,
"filename": "neurocode-1.0.0.tar.gz",
"has_sig": false,
"md5_digest": "24772f71bedbc4d014f0589400c4db72",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.9,<3.13",
"size": 27711,
"upload_time": "2023-11-19T15:49:06",
"upload_time_iso_8601": "2023-11-19T15:49:06.679462Z",
"url": "https://files.pythonhosted.org/packages/25/0a/58c4e297150550bd1f59a901a83a4296f761b7180294ec5795b77786cd63/neurocode-1.0.0.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2023-11-19 15:49:06",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "neurocode-ai",
"github_project": "neurocode",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"lcname": "neurocode"
}