|
12345678910111213141516171819202122232425262728293031323334353637 |
- function p = predict(Theta1, Theta2, X)
- %PREDICT Predict the label of an input given a trained neural network
- % p = PREDICT(Theta1, Theta2, X) outputs the predicted label of X given the
- % trained weights of a neural network (Theta1, Theta2)
-
- % Useful values
- m = size(X, 1);
- num_labels = size(Theta2, 1);
-
- % You need to return the following variables correctly
- p = zeros(size(X, 1), 1);
- X = [ones(m,1), X];
- % ====================== YOUR CODE HERE ======================
- % Instructions: Complete the following code to make predictions using
- % your learned neural network. You should set p to a
- % vector containing labels between 1 to num_labels.
- %
- % Hint: The max function might come in useful. In particular, the max
- % function can also return the index of the max element, for more
- % information see 'help max'. If your examples are in rows, then, you
- % can use max(A, [], 2) to obtain the max for each row.
- %
-
-
-
- a2 = sigmoid(X * Theta1');
- a2 = [ones(size(a2,1), 1), a2];
- p = sigmoid(a2 *Theta2');
- [temp_p,p] = max(p, [], 2);
-
-
-
-
- % =========================================================================
-
-
- end
|