Cursera: machine learning ex4.
This commit is contained in:
20
machine_learning/mlclass-ex4-008/mlclass-ex4/predict.m
Normal file
20
machine_learning/mlclass-ex4-008/mlclass-ex4/predict.m
Normal file
@@ -0,0 +1,20 @@
|
||||
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);
|
||||
|
||||
h1 = sigmoid([ones(m, 1) X] * Theta1');
|
||||
h2 = sigmoid([ones(m, 1) h1] * Theta2');
|
||||
[dummy, p] = max(h2, [], 2);
|
||||
|
||||
% =========================================================================
|
||||
|
||||
|
||||
end
|
||||
Reference in New Issue
Block a user