Diff of /README.md [000000] .. [dcdaea]

Switch to unified view

a b/README.md
1
来源:["合肥高新杯"心电人机智能大赛](https://tianchi.aliyun.com/competition/entrance/231754/introduction)
2
3
**A版单模型20190916在线F1-Score=0.801**
4
5
**B版单模型20191010在线F1-Score=0.837**
6
7
系统环境:*centos7 python3.6 pytorch1.0*
8
9
**大致思路:每个导联作为一个通道,使用1维卷积进行训练**
10
11
# 数据预处理
12
数据解压放在data目录下,使用8个导联的数据,简单进行train_val数据集划分
13
```shell
14
python data_process.py
15
```
16
17
# 模型训练
18
```shell
19
python main.py train #从零开始训练
20
```
21
22
# 模型测试
23
模型测试,在submit文件夹下生成提交结果
24
```shell
25
python main.py test --ckpt=..model_path #加载预训练权重进行测试
26
```
27
28
**一些细节**
29
30
 1. 本次测试模型为1dconv_resnet34,直接修改于torchvision
31
 2. 训练数据只进行了简单的数据增强,最终无normalize
32
 3. 由于设备问题,数据进行了重采样,推荐使用原数据
33
34
35
参考论文:
36
https://www.nature.com/articles/s41591-018-0268-3