fixing bug with hpdrpart DR-1142

vertica · Oct 15, 2015 · 9b43558 · 9b43558
1 parent bf1ed60
commit 9b43558
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/algorithms/HPdclassifier/R/hpdrpart.R b/algorithms/HPdclassifier/R/hpdrpart.R
@@ -271,8 +271,7 @@ hpdrpart <- function(formula, data, weights, subset , na.action = na.omit,
 	attr(model,"xlevels") <- attr(model,"xlevels")$Levels
 	names(attr(model,"xlevels")) <- colnames(data)[categorical_features]
 	response_col = colnames(data)[categorical_features] == response_name
-	response_col = which(response_col)
-	attr(model,"xlevels") <- attr(model,"xlevels")[-response_col]
+	attr(model,"xlevels") <- attr(model,"xlevels")[!response_col]
 	model$nExecutor = nExecutor
 	class(model) <- c("hpdrpart","rpart")