%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 objects with missing values are removed. When something else
% is desired, use one of the options in MISVAL. As some features have
% many missing values we recommand to remove these features using
% VAL = 'f-remove'.
%
% SEE ALSO <a href="http://37steps.com/prtools">PRTools Guide</a>, <a href="http://archive.ics.uci.edu/ml/">UCI Website</a>
% PRTOOLS, DATASETS, MISVAL

% Copyright: R.P.W. Duin, r.p.w.duin@prtools.org

function x = arrhythmia(val);

prdatasets(mfilename,1,'http://prtools.org/prdatasets/arrhythmia.dat');
if nargin<1
	val = 'remove';
end

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)];

a = load('arrhythmia.dat');
x = pr_dataset(a(:,1:(end-1)),a(:,end));
x = setfeatlab(x,fl);
x = setname(x,'Arrhythmia normal');
%x(:,14) = [];  % there are so many missing values, that I just remove this
%               % don't do this, nice but confusing
[x,msg] = misval(x,val);
user.desc = [user.desc msg];
x = setuser(x,user);
x = setlablist(x);
return
