%ARRHYTHMIA 420 objects with 279 features in 12 classes
%
% X = ARRHYTHMIA(CLASS,VAL)
%
% The aim is to distinguish between the presence and absence of cardiac
% arrhythmia and to classify it in one of the 16 groups. Class 01 refers
% to 'normal' ECG classes 02 to 15 refers to different classes of
% arrhythmia and class 16 refers to the rest of unclassified ones.
%
% X = ARRHYTHMIA(VAL);
%
% By default features with missing values are removed. When something else
% is desired, use one of the options in MISVAL for VAL.
%
% SEE ALSO PRTools Guide, UCI Website
% PRTOOLS, DATASETS, MISVAL
% Copyright: R.P.W. Duin
function x = arrhythmia(val)
if nargin < 1, val = 'f-remove'; end
a = pr_getdata('http://prtools.tudelft.nl/prdatasets/arrhythmia.dat',1);
user.desc=['The Arrhymthmia database from UCI. The aim is to ' ...
'distinguish between the presence and absence of cardiac ' ...
'arrhythmia and to classify it in one of the 16 groups.'];
user.link = 'ftp://ftp.ics.uci.edu/pub/machine-learning-databases/arrhythmia';
fl = {'age' 'sex' 'height' 'weight' 'QRS duration' 'P-R interval' ...
'Q-T interval' 'T interval' 'P interval' 'QRS' 'T' 'P' 'QRST' ...
'J' 'heartrate'};
fl1 = {'Q wave width' 'R wave width' 'S wave width' 'R'' wave width' ...
'S'' wave width' 'number of intrinsic deflections' ...
'ragged R wave' 'diphasic derivation of R wave' ...
'ragged P wave' 'diphasic derivation of P wave' ...
'ragged T wave' 'diphasic derivation of T wave'};
fl = [fl strcat('DI-',fl1) strcat('DII-',fl1) strcat('DIII-',fl1) ...
strcat('AVR-',fl1) strcat('AVL-',fl1) strcat('AVF-',fl1) ...
strcat('V1-',fl1) strcat('V2-',fl1) strcat('V3-',fl1) ...
strcat('V4-',fl1) strcat('V5-',fl1) strcat('V6-',fl1)];
fl2 = {'JJ wave ampl' 'Q wave ampl' 'R wave ampl' ...
'S wave ampl' 'R'' wave ampl' 'S'' wave ampl' 'P wave ampl' ...
'T wave ampl' 'QRSA sum' 'QRSTA'};
fl = [fl strcat('DI-',fl2) strcat('DII-',fl2) strcat('DIII-',fl2) ...
strcat('AVR-',fl2) strcat('AVL-',fl2) strcat('AVF-',fl2) ...
strcat('V1-',fl2) strcat('V2-',fl2) strcat('V3-',fl2) ...
strcat('V4-',fl2) strcat('V5-',fl2) strcat('V6-',fl2)];
x = pr_dataset(a(:,1:(end-1)),a(:,end));
x = setfeatlab(x,fl);
x = setname(x,'Arrhythmia normal');
x = setuser(x,user);
x = misval(x,val);
return