Fix bugs related to removal of class customization

jmgirard · May 14, 2018 · 8085924 · 8085924
1 parent ca04b8f
commit 8085924
Showing 1 changed file with 5 additions and 10 deletions.
diff --git a/mBINARY.m b/mBINARY.m
@@ -32,12 +32,7 @@
 %
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 
-if nargin < 3
-    POSCLASS = 1;
-    NEGCLASS = 0;
-end
-
-if nargin < 4
+if nargin < 2
     OUTPUT = 'table';
 end
 
@@ -55,10 +50,10 @@
 
 TEST = DATA(:, 1);
 CRIT = DATA(:, 2);
-TP = sum((TEST == POSCLASS) & (CRIT == POSCLASS));
-FP = sum((TEST == POSCLASS) & (CRIT == NEGCLASS));
-FN = sum((TEST == NEGCLASS) & (CRIT == POSCLASS));
-TN = sum((TEST == NEGCLASS) & (CRIT == NEGCLASS));
+TP = sum((TEST == 1) & (CRIT == 1));
+FP = sum((TEST == 1) & (CRIT == 0));
+FN = sum((TEST == 0) & (CRIT == 1));
+TN = sum((TEST == 0) & (CRIT == 0));
 
 ACC = (TP + TN) / (TP + TN + FP + FN);
 TPR = TP / (TP + FN);