function [wholeDataTrain, wholeDataTest, labelTrain, labelTest] = dataBaseLoader(categories, materials) wholeDataTrain = []; wholeDataTest = []; labelTrain = []; labelTest = []; for material=materials index = 0; for cat=categories load(['./rawdata/', num2str(material), '-', num2str(cat)]); temp = temp.'; lab = ones(size(temp,1),1).*index; % Cross varidation (train: 80%, test: 20%) cv = cvpartition(size(temp,1),'HoldOut',0.2); idx = cv.test; % Separate to training and test data dataTrain = temp(~idx,:); dataTest = temp(idx,:); wholeDataTrain = [wholeDataTrain; dataTrain]; labelTrain = [labelTrain; lab(~idx,:)]; wholeDataTest = [wholeDataTest; dataTest]; labelTest = [labelTest; lab(idx,:)]; index = index+1; end end disp('data Loaded') end