emotionPrediction.m

function pred_result = emotionPrediction(raw_data,t_labels)
% This funciton applied k-fold cross validation on the transferred data
% Input: raw_data - initial data
%        t_labels - target labels in column
% Output: pred_result  -  a matrix, first column is actual label, second
%                         column is predicted label (1~6)

% Constant
k = 10;  % 10 fold cross validation 
num_examples = size(raw_data,1);

% data partition
r = mod(num_examples,k);
set_sizes = zeros(1,k);
set_sizes(1,1:r) = floor(num_examples/k)+1;
set_sizes(1,r+1:end) = floor(num_examples/k);

% for every k-cross validation iteration, select indices from indxpool for
% training and validation sets
idxpool = randperm(num_examples);

% predict result for test data, first column is actual label, second column
% is predicted label
pred_result = zeros(num_examples,2);
tmp_result = zeros(num_examples,max(unique(t_labels)));

% k-fold cross validation (6 trees generated at each fold)
for i = 1:k 
    for em = 1:max(unique(t_labels))
        ft_lb = datatrans(raw_data,t_labels,em);
        % test set
        tst_start = sum(set_sizes(1:i-1))+1;
        tst_end = tst_start+set_sizes(i)-1;  
        tst_idx = idxpool(tst_start:tst_end);
        tst_data = ft_lb(tst_idx,1:end-1);
        
        % training set
        trn_idx = idxpool(setdiff(1:end,tst_start:tst_end));
        tree = ID3(ft_lb(trn_idx,:)); 
        
        % test
        for j = 1:size(tst_data,1)
            if (travelTree(tree,tst_data(j,:)))
                tmp_result(tst_idx(j),em) = 1;
                pred_result(tst_idx(j),1) = t_labels(tst_idx(j),end);
            end
        end
    end   
end

% if the example is classified to more than one class, randomly select one
% classified class as the result
for i = 1:size(tmp_result,1)
    if sum(tmp_result(i,:)) == 0
        pred_result(i,2) = 0;
    end
    
    if sum(tmp_result(i,:)) == 1
        pred_result(i,2) = find(tmp_result(i,:)==1);
    end
    
    if sum(tmp_result(i,:)) > 1
        idx = find(tmp_result(i,:)==1);
        pred_result(i,2) = idx(randperm(size(idx,2),1));
    end
end