function [trieda]=kNNclass(data)
%% nacitanie obrazov klasifikátora
load obrazy_trieda_1.mat
load obrazy_trieda_2.mat

trieda_1_ucenie=vysledky_trieda_1;
trieda_2_ucenie=vysledky_trieda_2;

%% suradnice vsetkych bodov v priestore
x=[trieda_1_ucenie(:,1)' trieda_2_ucenie(:,1)'];
y=[trieda_1_ucenie(:,2)' trieda_2_ucenie(:,2)'];
z=[trieda_1_ucenie(:,3)' trieda_2_ucenie(:,3)'];

%% nezaradeny obraz
nezaradeny=data;

%% E vzdialensot
E=sqrt((x-nezaradeny(1)).^2+(y-nezaradeny(2)).^2+(z-nezaradeny(3)).^2);

%% 5-NN klasifikator
k=4;
A=0; B=0;
for i=1:k
    [hodnota pozicia]=min(E);
    E(pozicia)=NaN;
    if pozicia<=size(trieda_1_ucenie,1)
        A=A+1;
    else
        B=B+1;
    end
end

if A>B
    trieda=1;
else
    trieda=2;
end