8000 Plot ica comparison by Ganasekhar-gif · Pull Request #13215 · mne-tools/mne-python · GitHub
[go: up one dir, main page]
Skip to content

Plot ica comparison #13215

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 10 commits into
base: main
Choose a base branch
from
Open

Plot ica comparison #13215

Changes from 2 commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
9b76bc0
update ica_comparison.py
Ganasekhar-gif Apr 15, 2025
dc72b25
create newfeature.rst
Ganasekhar-gif Apr 15, 2025
f4b0bd8
update ica_comparison.py
Ganasekhar-gif Apr 15, 2025
feaa0b6
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Apr 15, 2025
ad7b1a1
create 13215.enhancement.rst
Ganasekhar-gif Apr 15, 2025
a513556
Merge branch 'plot_ica_comparison' of https://github.com/Ganasekhar-g…
Ganasekhar-gif Apr 15, 2025
d94f2a5
Merge branch 'main' into plot_ica_comparison
Ganasekhar-gif Apr 21, 2025
bc55115
Merge branch 'main' into plot_ica_comparison
Ganasekhar-gif May 2, 2025
dae4983
Merge branch 'main' into plot_ica_comparison
Ganasekhar-gif May 7, 2025
422710c
Merge branch 'main' into plot_ica_comparison
Ganasekhar-gif Jun 3, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
34 changes: 14 additions & 20 deletions examples/preprocessing/ica_comparison.py
View file
Open in desktop
8000
Original file line number Diff line number Diff line change
Expand Up @@ -24,12 +24,14 @@

# %%

from pathlib import Path
from time import time

import numpy as np

import mne
from mne.preprocessing import ICA
from mne.datasets import sample
import numpy as np
from time import time
from pathlib import Path
from mne.preprocessing import ICA

print(__doc__)

Expand All @@ -42,7 +44,7 @@

# Load sample dataset
data_path = Path(sample.data_path())
raw_file = data_path / 'MEG' / 'sample' / 'sample_audvis_raw.fif'
raw_file = data_path / "MEG" / "sample" / "sample_audvis_raw.fif"
raw = mne.io.read_raw_fif(raw_file, preload=True)
raw.pick_types(meg=True, eeg=False, eog=True)
raw.crop(0, 60)
Expand Down Expand Up @@ -87,6 +89,7 @@ def run_ica(raw_input, method, fit_params=None, reject=None):

return ica, fit_time


# %%


Expand All @@ -107,9 +110,7 @@ def run_all_ica(raw_input, label, reject):
icas[full_label] = ica
fit_times[full_label] = t

eog_inds, _ = ica.find_bads_eog(
raw_input, threshold=3.0, verbose='ERROR'
)
eog_inds, _ = ica.find_bads_eog(raw_input, threshold=3.0, verbose="ERROR")
if eog_inds:
eog_components[full_label] = eog_inds[0]
print(f"{full_label}:Detected EOG comp at index {eog_inds[0]}")
Expand All @@ -119,16 +120,13 @@ def run_all_ica(raw_input, label, reject):

return icas, fit_times, eog_components


# %%


# Run on both raw versions
icas_clean, times_clean, eog_clean = run_all_ica(
raw_clean, "clean", reject_clean
)
icas_noisy, times_noisy, eog_noisy = run_all_ica(
raw_noisy, "noisy", reject_noisy
)
icas_clean, times_clean, eog_clean = run_all_ica(raw_clean, "clean", reject_clean)
icas_noisy, times_noisy, eog_noisy = run_all_ica(raw_noisy, "noisy", reject_noisy)

# Combine results
icas = {**icas_clean, **icas_noisy}
Expand All @@ -143,9 +141,7 @@ def run_all_ica(raw_input, label, reject):
comp = eog_comps.get(key)
if comp is not None:
icas[key].plot_components(
picks=[comp],
title=f"{key} - EOG Component (Clean Data)",
show=True
picks=[comp], title=f"{key} - EOG Component (Clean Data)", show=True
)

# %%
Expand All @@ -156,7 +152,5 @@ def run_all_ica(raw_input, label, reject):
comp = eog_comps.get(key)
if comp is not None:
icas[key].plot_components(
picks=[comp],
title=f"{key} - EOG Component (Noisy Data)",
show=True
picks=[comp], title=f"{key} - EOG Component (Noisy Data)", show=True
)
0