Revert "javapredict now checks probabilities rather than class labels"

This reverts commit e086490.
AlicTsai · Mar 13, 2015 · 22f9da2 · 22f9da2
1 parent e086490
commit 22f9da2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/tests/Utils/shared_javapredict_SRF.R b/R/tests/Utils/shared_javapredict_SRF.R
@@ -57,7 +57,7 @@ if (nrow(prediction1) != nrow(prediction2)) {
   stop("Number of rows mismatch")
 }
 
-match <- all(prediction1[,2:3] == prediction2[,2:3])
+match <- all(prediction1 == prediction2)
 if (! match) {
   for (i in 1:nrow(prediction1)) {
     rowmatches <- all(prediction1[i,] == prediction2[i,])