Predict function code

function p = predict(Theta1, Theta2, X)

m = size(X, 1);
num_labels = size(Theta2, 1);

p = zeros(size(X, 1), 1);

h1 = sigmoid([ones(m, 1) X] * Theta1');
h2 = sigmoid([ones(m, 1) h1] * Theta2');

for i =1:size(h2,1)
  if h2(i)>0.5
    p(i) = 1;
  else
    p(i) = 0;
  end
end

end