AIWIN 心电图智能诊断Baseline【线上0.719】

今天老肥和大家分享的是AIWIN的秋季赛-心电图智能诊断竞赛的任务一Baseline方案，线上与线下验证得分均为0.719，采用的是单模树模型。

赛题背景

心电图是临床最基础的一个检查项目，因为安全、便捷成为心脏病诊断的利器。每天都有大量的心电图诊断需求，但是全国范围内诊断心电图的专业医生数量不足，导致很多医院都面临专业心电图医生短缺的情况。

人工智能技术的出现，为改善医生人力资源不足的问题带来了全新的可能。由于心电图数据与诊断的标准化程度较高，相对较易于运用人工智能技术进行智能诊断算法的开发。

赛题任务与数据

本赛题任务一为要求针对心电图输出二元（正常 v.s 异常）分类标签，评价指标为F1，心电数据的单位为mV，采样率为 500HZ，记录时长为 10 秒，存储格式为 MAT；文件中存储了 12 导联的电压信号（包含了I，II，III，aVR，aVL，aVF，V1，V2，V3，V4，V5 和 V6）。

Baseline方案

每个样本心电图的数据的形状为（12，5000）， 一个非常直观的思路便是将其进行分导联采样再将所有导联展平作为训练的原始数据避免特征维度过大。

def get_data(sep_length):
    """
    读取数据，提取部分数据用于训练预测
    """

    train_path = glo