心律失常分类

使用CNN模型实现实现MIT-BIH的数据库分析，读取数据库数据，进行相关模型训练和测试

function [data_x target_y] = initInput（）
% This script extracts training data from the MIT-BIH Arrythmia Database
% 
% Output values returned are:
%   data_x: 4-D double containing the input data in the following form
%       [128 1 1 numberOfReadings]
%   target_y: 1-D Categorial containing the signal labels
%       [numberOfReadings 1]
%
% Author: 06/15/17 - by Arshan Hashemi
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

num = 1;
fs = 360; % sample rate for MIT-BIH Arrhytmia Database
samples = 646400; % samples for 30 minutes

% Training files to split data by recording （Test 3）
training_files = {‘mitdb/100‘ ‘mitdb/232‘‘mitdb/109‘ ‘mitdb/106‘...
    ‘mitdb/102‘ ‘mitdb/118‘ ‘mitdb/207‘‘mitdb/231‘ ‘mitdb/103‘...
    ‘mitdb/208‘ ‘mitdb/118‘ ‘mitdb/214‘ ‘mitdb/104‘ ‘mitdb/201‘...
    ‘mitdb/203‘ ‘mitdb/116‘ ‘mitdb/215‘};
    
% Training files to split data by time or random（Test 1 and 2）
%training_files = {‘mitdb/100‘ ‘mitdb/232‘‘mitdb/109‘ ‘mitdb/106‘...
 %   ‘mitdb/102‘ ‘mitdb/118‘ ‘mitdb/207‘‘mitdb/231‘ ‘mitdb/103‘...
  %  ‘mitdb/208‘ ‘mitdb/112‘ ‘mitdb/214‘ ‘mitdb/104‘ ‘mitdb/201‘...
   % ‘mitdb/203‘ ‘mitdb/116‘ ‘mitdb/215‘‘mitdb/101‘ ‘mitdb/107‘... 
    %‘mitdb/111‘ ‘mitdb/119‘ ‘mitdb/200‘ ‘mitdb/209‘ ‘mitdb/222‘...
   % ‘mitdb/212‘ ‘mitdb/217‘‘mitdb/124‘ ‘mitdb/115‘ ‘mitdb/213‘...
   % ‘mitdb/222‘ ‘mitdb/209‘ ‘mitdb/234‘ ‘mitdb/221‘ ‘mitdb/223‘...
   % ‘mitdb/209‘ ‘mitdb/114‘ ‘mitdb/108‘ ‘mitdb/121‘ ‘mitdb/123‘...
   % ‘mitdb/231‘ ‘mitdb/233‘};

j = 1;
% Preallocate
data = zeros（500 128）;
target（500） = char（0）;

for f = 1 : length（training_files）
filename = char（training_files（f））
[tm signal]=rdsamp（filename 1 samples）;
signal = signal（:1）;

[anntype~~]=rdann（filename‘atr‘[]samples）;

for k = 1 : size（ann1）
  if ann（k） <= samples
      stop = k;
  end
end

ann = ann（1 : stop）;
type = type（1 : stop）;

start = 1;
if （ann（1） < 64）
    start = 4;
end

%stop = stop - 5;
stop = ceil（stop/6）;

% 6 Types of beats: Normal Paced Left BBB PVC APC RIGHT BBB
%                   N / L V A R
k = start;
%for k = start : stop    
 while k < stop
    if type（k） == ‘N‘ || type（k） == ‘/‘ || type（k） == ‘L‘ ||...
            type（k） == ‘V‘ || type（k） == ‘A‘ || type（k） == ‘R‘
        for i = 1 : 128
            data（ji） = signal（ann（k） - 63 + i）;
        end
        data_x（:11j） = data（j :）;
        target（j） = type（k）;
        j = j + 1;
    end
    %increment k an extra time
    k = k + 1;
end
end

target_y = categorical（cellstr（target‘））;

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2017-10-10 00:25  CNN-arrhythmia-dection-master\
     文件         247  2017-10-10 00:25  CNN-arrhythmia-dection-master\README.md
     文件        2578  2017-10-10 00:25  CNN-arrhythmia-dection-master\initInput.m
     文件        1692  2017-10-10 00:25  CNN-arrhythmia-dection-master\initPatientSpecTest.m
     文件        3324  2017-10-10 00:25  CNN-arrhythmia-dection-master\initPatientSpecific.m
     文件        2419  2017-10-10 00:25  CNN-arrhythmia-dection-master\initTest.m
     文件        1085  2017-10-10 00:25  CNN-arrhythmia-dection-master\launchNetwork.m
     文件         652  2017-10-10 00:25  CNN-arrhythmia-dection-master\testNetwork.m