fixed coding function issue in nytro_test and added new coding function

LCSL · Oct 29, 2015 · 601945c · 601945c
1 parent 41c1904
commit 601945c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/codingFunctions/binaryClassification/zeroOneBin.m b/codingFunctions/binaryClassification/zeroOneBin.m
@@ -0,0 +1,12 @@
+function [ Ypred ] = zeroOneBin( Yscores )
+%zeroOneBin Encodes prediction vector as: Class 1: +1; Class 2: 0
+%   Detailed explanation goes here
+
+    if size(Yscores,2)>1
+        error('zeroOneBin is designed for binary classification coding');
+    end
+
+    Ypred = zeros(size(Yscores,1),1);
+    Ypred(Yscores>0.5) = 1;
+end
+
diff --git a/nytro_test.m b/nytro_test.m
@@ -13,7 +13,7 @@
     tic
     output.YtePred = KnmTe * training_output.best.alpha;
     if ~isempty(training_output.config.crossValidation.codingFunction)
-        output.YtePred = codingFunction(output.YtePred);
+        output.YtePred = training_output.config.crossValidation.codingFunction(output.YtePred);
     end
     output.time.prediction = toc;