From 42cbbc515e689058b58b5c68d9e485611794b108 Mon Sep 17 00:00:00 2001 From: Jan Stanstrup Date: Wed, 11 Nov 2015 12:54:20 +0100 Subject: [PATCH] Handle NA vaules in RemoveHighlyCorrelatedFeatures added use="pairwise.complete.obs" for the correlation matrix in RemoveHighlyCorrelatedFeatures --- camb/R/Modeling.R | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/camb/R/Modeling.R b/camb/R/Modeling.R index 886f86d..9a10199 100755 --- a/camb/R/Modeling.R +++ b/camb/R/Modeling.R @@ -53,7 +53,7 @@ RemoveNearZeroVarianceFeatures <- function(dataset, frequencyCutoff = 30/1,...) } RemoveHighlyCorrelatedFeatures <- function(dataset, correlationCutoff = 0.95,...) { - hc.columns <- findCorrelation(cor(dataset$x.train), correlationCutoff,...) + hc.columns <- findCorrelation(cor(dataset$x.train,use="pairwise.complete.obs"), correlationCutoff,...) if (length(hc.columns) != 0) { message(paste(length(hc.columns), "features removed with correlation above cutoff")) dataset$x.train <- dataset$x.train[, -hc.columns]