minor tweak logistic regression

bakisit · Mar 9, 2017 · 9e70e29 · 9e70e29
1 parent 5074a11
commit 9e70e29
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/chapter04/logitBinPred.m b/chapter04/logitBinPred.m
@@ -9,6 +9,6 @@
 % Written by Mo Chen ([email protected]).
 X = [X;ones(1,size(X,2))];
 w = model.w;
-p = exp(-log1pexp(-w'*X)); 
+p = sigmoid(w'*X);
 y = round(p);
 
diff --git a/chapter04/logitMnPred.m b/chapter04/logitMnPred.m
@@ -9,6 +9,5 @@
 % Written by Mo Chen ([email protected]).
 W = model.W;
 X = [X; ones(1,size(X,2))];
-A = W'*X;                                   
-P = exp(bsxfun(@minus,A,logsumexp(A,1)));  
+P = softmax(W'*X);
 [~, y] = max(P,[],1);