use 0-1 error instead of squared error

li-xin-yi · Feb 18, 2015 · 7d23b05 · 7d23b05
1 parent 0177ebf
commit 7d23b05
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/quiz4/predict_check_nnet.m b/quiz4/predict_check_nnet.m
@@ -2,8 +2,7 @@
  re=0;
   N=length(test);
      for i=1:N
-        predict=tanh(w2'*[1;tanh(w1'*[1,test(i,1:2)]')]);
-        re=re+(test(i,3)-predict)^2;
+        predict(i,1)=sign(tanh(w2'*[1;tanh(w1'*[1,test(i,1:2)]')]));
      end
-     re=re/N;
+     re=sum(sign(abs(predict-test(:,3))))/N;
 end