cost.m 961 B

1234567891011121314151617181920212223242526
  1. function [ J ] = cost( nn_params, ...
  2. input_layer_size, ...
  3. hidden_layer_size, ...
  4. num_labels, ...
  5. X, y, lambda )
  6. Theta1 = reshape(nn_params(1:hidden_layer_size * (input_layer_size + 1)), ...
  7. hidden_layer_size, (input_layer_size + 1));
  8. Theta2 = reshape(nn_params((1 + (hidden_layer_size * (input_layer_size + 1))):end), ...
  9. num_labels, (hidden_layer_size + 1));
  10. mapy = zeros(length(y), num_labels);
  11. for i = 1:length(y)
  12. mapy(i, y(i)) = 1;
  13. end
  14. a1 = [ones(size(X, 1),1) X]';
  15. z2 = Theta1 * a1;
  16. a2 = [ones(1, size(z2,2)); sigmoid(z2)];
  17. z3 = Theta2 * a2;
  18. a3 = sigmoid(z3);
  19. n = size(a3, 1);
  20. J = mean(sum( - mapy' .* log(a3) + (mapy' -1) .* log(1 - a3)));
  21. J = J + lambda * 0.5 / size(X, 1) * (sum(sum(Theta1(:,2:end) .^2)) + sum(sum(Theta2(:,2:end) .^2)));
  22. end