%Setup parameters
data_length = 3072;
num_classes = 10;
svm_buffer = 1;
step_size = 0.01;
num_iters = 50;
%Load CIFAR-10 data
load('data_batch_1.mat');

%Might want to make these random values smaller
random_W = rand(num_classes, data_length);

online_W = onlineDescent(random_W,data_length,num_classes,svm_buffer,step_size,data,labels);
batch_W = batchDescent(random_W,data_length,num_classes,svm_buffer,step_size,data,labels,num_iters);
combined_W = batchDescent(online_W,data_length,num_classes,svm_buffer,step_size,data,labels,num_iters);

%Load CIFAR-10 test data
load('test_batch.mat');

%Test the weights
data = double(data);
test_scores = batch_W * data';
[M, I] = max(test_scores,[],1);

evaluation = I-(double(labels')+1);
idx = evaluation==0;
num_corr_batch = sum(idx);
corr_perc_batch = num_corr_batch / length(labels');