# EmbedSleepNet
This repository reproduces TinySleepNet for sleep stage prediction based on the signal channel EEG
using PyTorch and implements a new smaller and faster network called EmbedSleepNet.
# Environment Setup
1. Create a virtual environment with Python 3.8:
`virtualenv venv --python=python3.8`
2. Activate the environment:
`source venv/bin/activate` or `venv\Scripts\activate.bat` (for Windows)
3. Install dependencies:
`pip install -r requirements.txt`
4. Download and extract Sleep-EDF dataset from https://www.physionet.org/content/sleep-edfx/1.0.0/
5. Preprocess the data by running `python preprocess.py --data_path PATH_TO_sleep-cassete_FOLDER`
# Running
To start the training process simply run:
`python train.py --flavor=[embed, tiny]`, where tiny represents original TinySleepNet,
and embed represents newly introduced EmbedSleepNet.
You may additionally specify the number of epochs and model output name, for example:
`python train.py --flavor embed --epochs 450 --model-name model`
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 基于注意力机制的单通道EEG睡眠分期项目python源码.zip 【资源说明】 1、该资源内项目代码都是经过测试运行成功,功能正常的情况下才上传的,请放心下载使用。 2、适用人群:主要针对计算机相关专业(如计科、信息安全、数据科学与大数据技术、人工智能、通信、物联网、数学、电子信息等)的同学或企业员工下载使用,具有较高的学习借鉴价值。 3、不仅适合小白学习实战练习,也可作为大作业、课程设计、毕设项目、初期项目立项演示等,欢迎下载,互相学习,共同进步!
资源推荐
资源详情
资源评论
收起资源包目录
基于注意力机制的单通道EEG睡眠分期项目python源码.zip (20个子文件)
projectcode30312
README.md 81B
EmbedSleepNet
preprocess.py 5KB
benchmark.py 368B
lightning_wrapper.py 3KB
dataset.py 3KB
model.py 3KB
.idea
misc.xml 199B
inspectionProfiles
Project_Default.xml 942B
profiles_settings.xml 174B
modules.xml 288B
EmbedSleepNet-main.iml 409B
.gitignore 182B
requirements.txt 179B
.gitignore 2KB
train.py 1KB
__pycache__
model.cpython-38.pyc 4KB
dataset.cpython-38.pyc 3KB
lightning_wrapper.cpython-38.pyc 4KB
README.md 1007B
model_search.py 723B
共 20 条
- 1
资源评论
龙年行大运
- 粉丝: 1006
- 资源: 3852
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功