-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpredict.m
28 lines (19 loc) · 824 Bytes
/
predict.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
function p = predict(theta, X)
%PREDICT Predict whether the label is 0 or 1 using learned logistic
%regression parameters theta
% p = PREDICT(theta, X) computes the predictions for X using a
% threshold at 0.5 (i.e., if sigmoid(theta'*x) >= 0.5, predict 1)
m = size(X, 1); % Number of training examples
% You need to return the following variables correctly
p = zeros(m, 1);
% ====================== YOUR CODE HERE ======================
% Instructions: Complete the following code to make predictions using
% your learned logistic regression parameters.
% You should set p to a vector of 0's and 1's
%
pos = find(sigmoid(X*theta) >= 0.5);
neg = find(sigmoid(X*theta) < 0.5);
p(pos) = 1;
p(neg) = 0;
% =========================================================================
end