Fixed testing bug

topepo · Jun 26, 2017 · 68fbe19 · 68fbe19
1 parent 37f226c
commit 68fbe19
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/RegressionTests/Code/ctree.R b/RegressionTests/Code/ctree.R
@@ -62,12 +62,9 @@ set.seed(849)
 test_class_rec <- train(recipe = rec_cls,
                         data = training,
                         method = "ctree", 
-                        trControl = cctrl1,
-                        metric = "ROC")
+                        trControl = cctrl1)
 
 test_class_pred_rec <- predict(test_class_rec, testing[, -ncol(testing)])
-test_class_prob_rec <- predict(test_class_rec, testing[, -ncol(testing)], 
-                               type = "prob")
 
 test_levels <- levels(test_class_cv_model)
 if(!all(levels(trainY) %in% test_levels))

diff --git a/RegressionTests/Code/ctree2.R b/RegressionTests/Code/ctree2.R
@@ -63,12 +63,9 @@ set.seed(849)
 test_class_rec <- train(recipe = rec_cls,
                         data = training,
                         method = "ctree2", 
-                        trControl = cctrl1,
-                        metric = "ROC")
+                        trControl = cctrl1)
 
 test_class_pred_rec <- predict(test_class_rec, testing[, -ncol(testing)])
-test_class_prob_rec <- predict(test_class_rec, testing[, -ncol(testing)], 
-                               type = "prob")
 
 test_levels <- levels(test_class_cv_model)
 if(!all(levels(trainY) %in% test_levels))