脑电分析系列[MNE-Python-5]| Python机器学习算法随机森林判断睡眠类型
案例介绍
本案例通过对多导睡眠图(Polysomnography,PSG)数据进行睡眠阶段的分类来判断睡眠类型。
训练:对Alice的睡眠数据进行训练;
测试:利用训练结果对Bob的睡眠数据进行测试,判断其睡眠类型。
在分析之前,先简单介绍一下多导睡眠图
多导睡眠图(Polysomnography,PSG)又称睡眠脑电图。主要用于睡眠和梦境研究以及抑郁症和睡眠呼吸暂停综合征的诊断。
多导睡眠图是通过不同部位的生物电或通过不同传感获得生物讯号,经前置放大,输出为不同的电讯号,记录出不同的图形以供分析。
数据集介绍
本案例用的数据是来自于PhysioNet上关于健康受试者的年龄对睡眠影响研究的公开数据集的一个子集。
mne.datasets.sleep_physionet.age.fetch_data可以下载PhysioNet数据集的子数据集。
该子数据集中包含20位受试者的实验数据,记录当时年龄为25-34岁的10位男性和10位女性的实验数据。由于受试者13的第二个记录遗失了,所以除了受试者13以外,每个受试者都有两次夜间记录。
Sleep Physionet数据集使用8个标签进行标注,代表8各阶段:
Wake (W),
Stage 1,
Stage 2,
Stage 3,
Stage 4,
REM(R),
Movement time(M),
Stage(?). Stage(?)-(not scored)
唤醒-Wake(W)、第1阶段、第2阶段、第3阶段、第4阶段、对应于从轻度睡眠到深度睡眠的范围;REM睡眠(R),其中REM是Rapid Eye的缩写,表示快速眼运动睡眠,运动(M)和阶段(?)的任何未得分部分。
第一步:导入工具库
import numpy as np
import matplotlib.pyplot as pltimport mne
from mne.datasets.sleep_physionet.age import fetch_data
from mne.time_frequency import psd_welchfrom sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import accuracy_score
from sklearn.metrics import confusion_matrix
from sklearn.metrics import classification_report
from sklearn.pipeline import make_pipeline
from sklearn.preprocessing import FunctionTransformer
第二步:加载数据
在这里,我们从两个主题下载数据,最终目标是获得时间片段(epochs)及其相关的地面真理。
MNE为我们提供了mne.datasets.sleep_physionet.age.fetch_data(),可以方便地从Sleep Physionet数据集下载数据。
给定主题和记录的列表,提取程序将下载数据并为每个主题提供数据, 一对文件:
-PSG.edf包含多导睡眠图。来自EEG头盔的原始数据,
-Hypnogram.edf包含专家记录的注释。
然后,将这两个对象合并到mne.io.Raw对象中,就可以根据注释的描述提取事件以获得时间片段(epochs)。
"""
可以通过
mne.datasets.sleep_physionet.age.fetch_data(subjects,recording,path)
来获取PhysioNet多导睡眠图数据集文件。
subjects:表示想要使用哪些受试者对象,可供选择的受试者对象范围为0-19。
recording:表示夜间记录的编号(索引),有效值为:[1]、[2]或[1、2]。path:PhysioNet数据的存放地址,如果没有给定,则加载默认存放数据的地址;
如果默认存放数据集的地址不存在数据,则从网络中下载相关数据。
"""
# 选择两个受试者实验对象ALICE, BOB(该名字并非实验中的真实名,这里是为了方便才临时取的名字)
ALICE, BOB = 0, 1
# 加载ALICE, BOB的实验数据文件
[alice_files, bob_files] = fetch_data(subjects=[ALICE, BOB], recording=[1])# 通道名称映射
mapping = {'EOG horizontal': 'eog','Resp oro-nasal': 'misc','EMG submental': 'misc','Temp rectal': 'misc','Event marker': 'misc'}#读取ALICE的edf文件,和其对应的注释文件
raw_train = mne.io.read_raw_edf(alice_files[0])
annot_train = mne.read_annotations(alice_files[1])raw_train.set_annotations(annot_train, emit_warning=False)
raw_train.set_channel_types(mapping)# 绘制空0s开始,时间窗口长度为40s的连续通道数据波形图
raw_train.plot(duration=40, scalings='auto')
plt.show()
这里仅使用5个阶段:唤醒(W),阶段1,阶段2,阶段3/4和REM睡眠(R)。
为此,这里使用mne.events_from_annotations()中的event_id参数来选择我们感兴趣的事件,并将事件标识符与每个事件相关联。
"""
睡眠表示与事件映射
"""
annotation_desc_2_event_id = {'Sleep stage W': 1,'Sleep stage 1': 2,'Sleep stage 2': 3,'Sleep stage 3': 4,'Sleep stage 4': 4,'Sleep stage R': 5}events_train, _ = mne.events_from_annotations(raw_train, event_id=annotation_desc_2_event_id, chunk_duration=30.)# 创建一个新的event_id以统一 阶段3和4
event_id = {'Sleep stage W': 1,'Sleep stage 1': 2,'Sleep stage 2': 3,'Sleep stage 3/4': 4,'Sleep stage R': 5}# 绘制事件数据
mne.viz.plot_events(events_train, event_id=event_id,sfreq=raw_train.info['sfreq'])# 保留颜色代码以便进一步绘制
stage_colors = plt.rcParams['axes.prop_cycle'].by_key()['color']
根据注释中的事件从数据创建epochs(时间片段)
tmax = 30. - 1. / raw_train.info['sfreq'] # tmax in included
"""
所创建的是时间从tmin=0开始,到tmax为止的epochs
"""
epochs_train = mne.Epochs(raw=raw_train, events=events_train,event_id=event_id, tmin=0., tmax=tmax, baseline=None)print(epochs_train)
第三步:加载Bob的数据作为测试数据
按照上述相同的步骤来获取Bob的测试数据
raw_test = mne.io.read_raw_edf(bob_files[0])
annot_test = mne.read_annotations(bob_files[1])
raw_test.set_annotations(annot_test, emit_warning=False)
raw_test.set_channel_types(mapping)
events_test, _ = mne.events_from_annotations(raw_test, event_id=annotation_desc_2_event_id, chunk_duration=30.)
epochs_test = mne.Epochs(raw=raw_test, events=events_test, event_id=event_id,tmin=0., tmax=tmax, baseline=None)print(epochs_test)
特征工程
观察不同睡眠阶段的功率谱密度(PSD)图,可以看到不同睡眠阶段具有不同的特征。这些签名在Alice和Bob的数据中保持相似。
在本节的其余部分中,将基于特定频带中的相对功率来创建EEG特征,以捕获数据中睡眠阶段之间的差异。
fig, (ax1, ax2) = plt.subplots(ncols=2)# iterate over the subjects
stages = sorted(event_id.keys())
for ax, title, epochs in zip([ax1, ax2],['Alice', 'Bob'],[epochs_train, epochs_test]):for stage, color in zip(stages, stage_colors):epochs[stage].plot_psd(area_mode=None, color=color, ax=ax,fmin=0.1, fmax=20., show=False,average=True, spatial_colors=False)ax.set(title=title, xlabel='Frequency (Hz)')
ax2.set(ylabel='uV^2/hz (dB)')
ax2.legend(ax2.lines[2::3], stages)
plt.tight_layout()
plt.show()
第四步:设计scikit-learn 转换器
创建一个函数,根据特定频带中的相对功率提取脑电图特征,从而能够根据脑电图信号预测睡眠阶段。
def eeg_power_band(epochs):"""脑电相对功率带特征提取该函数接受一个""mne.Epochs"对象,并基于与scikit-learn兼容的特定频带中的相对功率创建EEG特征。Parameters----------epochs : EpochsThe data.Returns-------X : numpy array of shape [n_samples, 5]Transformed data."""# 特定频带FREQ_BANDS = {"delta": [0.5, 4.5],"theta": [4.5, 8.5],"alpha": [8.5, 11.5],"sigma": [11.5, 15.5],"beta": [15.5, 30]}psds, freqs = psd_welch(epochs, picks='eeg', fmin=0.5, fmax=30.)# 归一化 PSDspsds /= np.sum(psds, axis=-1, keepdims=True)X = []for fmin, fmax in FREQ_BANDS.values():psds_band = psds[:, :, (freqs >= fmin) & (freqs < fmax)].mean(axis=-1)X.append(psds_band.reshape(len(psds), -1))return np.concatenate(X, axis=1)
第五步:根据 Alice的数据来预测Bob的睡眠阶段
使用scikit-learn进行多分类
下面展示了解决如何从爱丽丝的数据中预测鲍勃的睡眠阶段并尽可能避免重复样板代码的问题。
这里将利用sckit-learn的Pipeline和FunctionTransformer。
扩展:[Pipeline可以将许多算法模型串联起来,可以用于把多个estamitors级联成一个estamitor,比如将特征提取、归一化、分类组织在一起形成一个典型的机器学习问题工作流。FunctionTransformer将python函数转换为与estamitor兼容的对象。]
pipe = make_pipeline(FunctionTransformer(eeg_power_band, validate=False),RandomForestClassifier(n_estimators=100, random_state=42))
# 训练
y_train = epochs_train.events[:, 2]
pipe.fit(epochs_train, y_train)# 预测
y_pred = pipe.predict(epochs_test)# 评估准确率
y_test = epochs_test.events[:, 2]
acc = accuracy_score(y_test, y_pred)print("Accuracy score: {}".format(acc))
预测的准确精度为84.7%
查看分类报告做进一步分析
print(classification_report(y_test, y_pred, target_names=event_id.keys()))
从分类报告中可以看出,
Bob的每个阶段训练测试样本,以及对应的睡眠阶段的精度。
比如W阶段的精度为86%,测试样本为1856。测试总样本为2802。也可以看到其他一些指标比如召回率和F1值。
不用于商业行为,转载请联系后台
若有侵权,请后台留言,管理员即时删侵!
更多阅读
懂车Talks:脑机接口技术正在改变未来出行方式
脑-脑接口:人类大脑利用意念控制老鼠走迷宫
脑电分析系列4| MNE中数据结构Evoked及其对象创建
脑电分析系列3| MNE中数据结构Epoch及其创建方法
脑电分析系列2| MNE中数据结构Raw及其用法简介(更新)
投稿通道
如何让你的工作让更多人知晓和受益?
脑机接口社区就是这样一个连接学界、
企业界和爱好者的平台渠道。
社区鼓励高校实验室、企业或个人在我们平台上分享优质内容。
稿件要求
稿件系个人原创作品,若已在其他平台发表,请明确标注。
稿件一经录取,便提供稿费!
投稿通道
微信扫码,备注:投稿+姓名+单位
微信交流群,请扫码上方微信
(备注:姓名+单位+专业/领域行业)
QQ交流群:913607986
你的每一次在看,我都很在意!
脑电分析系列[MNE-Python-5]| Python机器学习算法随机森林判断睡眠类型相关推荐
- Python-EEG工具库MNE中文教程(5)-机器学习算法随机森林判断睡眠类型
目录 案例介绍 数据集介绍 第一步:导入工具库 第二步:加载数据 第三步:加载Bob的数据作为测试数据 特征工程 第四步:设计scikit-learn 转换器 第五步:根据 Alice的数据来预测Bo ...
- 脑机接口基础学习06---机器学习算法随机森林判断睡眠类型
案例介绍 本案例对多导睡眠图(Polysomnography,PSG)数据进行睡眠阶段的分类来判断睡眠类型 训练:对Alice的睡眠数据进行训练 测试:利用训练结果对Bob的睡眠数据进行测试,判断其睡 ...
- 脑电分析系列[MNE-Python-6]| Python读取.set文件和.locs文件
本案例主要介绍如何读取eeglab示例文件.set以及.locs文件. .set文件记录的是采集的信息,主要内容包括通道数,事件数,开始时间与结束时间等. .locs文件记录的是电极位置信息. 知识点 ...
- 脑电分析系列 | MNE-Python汇总
1 脑电分析系列MNE-Python教程 [MNE-1]| MNE-Python详细安装与使用(更新) [MNE-2]| MNE中数据结构Raw及其用法简介(更新) [MNE-3]| MNE中数据结构 ...
- 脑电分析系列[MNE-Python-11]| 信号空间投影SSP 应用
信号空间投影(SSP) 在前面一篇分享(脑电分析系列[MNE-Python-10]| 信号空间投影SSP数学原理)中提到,投影矩阵将根据您试图投射出的噪声种类而变化.信号空间投影(SSP)是一种通过比 ...
- 脑电分析系列[MNE-Python-13]| bad通道介绍
bad 通道 本教程主要介绍手动标记坏通道以及基于其他传感器的好信号重建"bad"通道. 导入工具库 mport os from copy import deepcopy impo ...
- 脑电分析系列[MNE-Python-9]| 参考电极应用
参考电极 MNE-Python中的平均参考设置 set_eeg_reference(self, ref_channels='average',projection=False, ch_type='au ...
- 脑电分析系列[MNE-Python-8]| 参考电极简介
参考电极 安置在头皮上的电极为作用电极(active electrode).放置在身体相对零电位点的电极即为参考电极(reference electrode),也称为参考电极或标准电极. 记录到的脑电 ...
- 脑电分析系列[MNE-Python-16]| 脑电数据的Epoching处理
import os.path as op import numpy as np import mne import matplotlib.pyplot as plt 在MNE中,epochs是指单个试 ...
最新文章
- 需求用例分析之六:业务用例之科伯恩系
- configure脚本分析
- 如何避免重要需求遗漏?
- [翻译]CryEngine3中光照的美术提示
- Using Custom Assemblies with Reports
- Hexo为文章设置目录与标签的方法
- 数据治理需要注意哪些问题
- meta 标签的作用
- spring cloud gateway 源码解析(4)跨域问题处理
- 使用Adorner显示WPF控件的边界点
- 日本艺伎的拍照姿势,竟源于孔子?看完真的涨知识···
- hexo next auto_excerpt无法使用
- 聊天系统:服务端 + 客户端 + web端
- 前端vue异形轮播图案例(带源码)
- 如何用计算机算p,App Store 上的“计算器++P”
- Android客户端与PC服务器如何实现Socket通信
- svn访问版本库时一直提示: please wait while the repository browser is initializing
- python pandas拆分单元格
- hexo-theme-next @modified LRH
- java 创建word文件_Java 创建Word
热门文章
- 如何解决make: Nothing to be done for `all‘ 的方法
- python异常之ModuleNotFoundError: No module named ‘test01inner02‘
- 微信小程序 textarea 简易解决方案
- Swift:print()vs println()vs NSLog()
- “参数”vs“参数”[重复]
- npm package.json文件中的依赖关系,devDependencies和peerDependencies之间有什么区别?
- 如何删除Android上ListViews之间的行?
- iOS Swift UISearchController的取消按钮
- 生产环境一个like模糊匹配SQL优化
- Vue开源项目库汇总