Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Hypo
candock
提交
848855c2
C
candock
项目概览
Hypo
/
candock
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
C
candock
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
848855c2
编写于
5月 03, 2019
作者:
H
hypox64
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Updata readme
上级
83c4050c
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
10 addition
and
10 deletion
+10
-10
README.md
README.md
+4
-4
how_to_run.md
how_to_run.md
+6
-6
未找到文件。
README.md
浏览文件 @
848855c2
# candock
这是一个用于记录毕业设计的日志仓库,其目的是尝试多种不同的深度神经网络结构(如LSTM,RESNET,DFCNN等)对单通道EEG进行自动化睡眠阶段分期.我们相信这些代码同时可以用于其他生理信号(如ECG,EMG等)的分类.希望这将有助于您的研究.
<br>
![
image
](
https://github.com/HypoX64/candock/blob/master/image/compare.png
)
## 如何运行
如果你需要运行这些代码(训练自己的模型或者使用预训练模型进行测试)请进入以下页面
<br>
[
How to run codes
](
https://github.com/HypoX64/candock/blob/master/how_to_run.md
)
<br>
...
...
@@ -30,7 +31,7 @@
*
multi_scale_resnet_1d 网络结构
<br>
该网络参考
[
geekfeiw / Multi-Scale-1D-ResNet
](
https://github.com/geekfeiw/Multi-Scale-1D-ResNet
)
<br>
修改后的
[
网络结构
如图
](
https://github.com/HypoX64/candock/blob/master/image/multi_scale_resnet_1d_network.png
)
<br>
修改后的
[
网络结构
](
https://github.com/HypoX64/candock/blob/master/image/multi_scale_resnet_1d_network.png
)
<br>
*
关于交叉验证
<br>
为了便于与其他文献中的方法便于比较,使用了两种交叉验证方法
<br>
...
...
@@ -42,14 +43,13 @@
accuracy = (TP+TN)/(TP+FN+TN+FP)
<br>
recall = sensitivity = (TP)/(TP+FN)
<br>
对于总体:
<br>
Top1.err.
Top1.err.
<br>
*
关于代码
<br>
目前的代码仍然在不断修改与更新中,不能确保其能工作.详细内容将会在毕业设计完成后抽空更新.
<br>
## 部分实验结果
该部分将持续更新... ...
<br>
[
[Confusion matrix]
](
https://github.com/HypoX64/candock/blob/master/
image/
confusion_mat
)
<br>
[
[Confusion matrix]
](
https://github.com/HypoX64/candock/blob/master/confusion_mat
)
<br>
#### 5-Fold Cross-Validation Results
*
sleep-edf
<br>
...
...
how_to_run.md
浏览文件 @
848855c2
...
...
@@ -2,7 +2,7 @@
-
Linux, Windows,mac
-
CPU or NVIDIA GPU + CUDA CuDNN
-
Python 3.5+
-
Pytroch 1.0
+
-
Pytroch 1.0
## Dependencies
This code depends on torchvision, numpy, scipy, h5py, matplotlib, mne , requests, hashlib, available via pip install.
<br>
...
...
@@ -23,20 +23,20 @@ git clone https://github.com/HypoX64/candock
cd
candock
```
### Train
*
d
ownload datasets
*
D
ownload datasets
```
bash
python3 download_dataset.py
```
*
choose
your options and run
*
Input
your options and run
```
bash
python3 train.py
--dataset_dir
'./datasets/sleep-edfx/'
--dataset_name
sleep-edfx
--signal_name
'EEG Fpz-Cz'
--sample_num
10
--model_name
lstm
--batchsize
64
--network_save_freq
5
--epochs
20
--lr
0.0005
--BID
5_95_th
--select_sleep_time
--cross_validation
subject
```
*
Notes
<br>
If want to use cpu to train or test, please
use
--no_cuda
If want to use cpu to train or test, please
input
--no_cuda
### Simple Test
*
download pretrained model & simple test data
[
Google Drive
](
https://drive.google.com/open?id=1pup2_tZFGQQwB-hoXRjpMxiD4Vmpn0Lf
)
*
choose
your options and run
*
Download pretrained model & simple test data
[
[Google Drive]
](
https://drive.google.com/open?id=1pup2_tZFGQQwB-hoXRjpMxiD4Vmpn0Lf
)
*
Input
your options and run
```
bash
python3 simple_test.py
```
\ No newline at end of file
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录