欢迎光临散文网 会员登陆 & 注册

信号处理--基于EEG脑电信号深度学习情绪分类 (同步)

2023-04-01 15:09 作者:大熊的二大爷  | 我要投稿

本文为一个信号处理专题的课程项目,主要是基于人体脑电信号,通过使用深度学习,来快速精准的识别被试的情绪。实验数据为私有数据集。情绪分为积极,中性,消极三种类别。该方法最后和传统朴素贝叶斯,支持向量机,logistic回归,决策树和随机森林分类器进行比较。


1 加载主要库函数


2 检查EEG脑电信号和数据预处理

  

 2.1 绘制不同种类数据大小比例分布图

  

不同种类数据大小分布

 

2.2 显示积极情绪的脑电信号

  

积极情绪脑电信号


2.3 显示消极情绪的脑电信号

 


消极情绪脑电信号

 2.4 显示中性情绪的脑电信号

 

 2.5 数据的预处理

 

  

3 搭建LSTM深度学习模型


3.1 定义模型的构建函数

 

3.2 构建模型

 

LSTM模型结构

3.3 模型训练和测试

  

LSTM模型训练

 

 3.4 使用confusion matrix 评估模型

 

LSTM模型分类混沌矩阵



 

传统模型分类混沌矩阵

 

LSTM模型分类报告

 

5 绘制模型训练loss和准确率

 

6 小结

本文使用了LSTM模型,在tensorflow开发环境,实现了EEG脑电情绪分类,并和传统的方法进行比较。


同步于CSDN



信号处理--基于EEG脑电信号深度学习情绪分类 (同步)的评论 (共 条)

分享到微博请遵守国家法律