R-caret: как использовать веса классов вместе с downSample для решения проблемы дисбаланса классов? - PullRequest
0 голосов
/ 24 октября 2019

У меня есть чрезвычайно несбалансированный набор данных. Чтобы справиться с этой проблемой, я пробовал отдельно разные методы дисбаланса классов: downSample , веса классов , настройка порога . Среди них пороговая настройка была наименее эффективной. Используя только downSample или веса классов, мне не удалось получить достаточно хорошие результаты: либо слишком много FalsePositives, либо FalseNegatives. Поэтому я хотел бы объединить эти две техники. Вот что я устал:

# produce some re-producible imbalanced data
set.seed(12345)
y <- as.factor(sample(c("M", "F"),
                      prob = c(0.1, 0.9),
                      size = 10000,
                      replace = TRUE))


x <- rnorm(10000)


DATA <- data.frame(y = as.factor(y), x)

set.seed(12345)
folds <- createFolds(dataSet$y, k = 10, 
                     list = TRUE, returnTrain = TRUE)

# class weights 
k <- 0.5
classWeights <- ifelse(DATA$y == "M",
                       (1/table(DATA$y)[1]) * k,
                       (1/table(DATA$y)[2]) * (1-k))

, поэтому, когда я не помещаю аргумент sampling в controlTrain:

# select algorithm
algorithm <- "bayesglm"

# train parameters
set.seed(12345)
traincontrol <- trainControl(method = "loocv", # resampling method
                             number = 10,
                             index = folds,
                             classProbs = TRUE, 
                             summaryFunction = twoClassSummary,
                             savePredictions = TRUE,
                             # sampling = "down"
                             )

fitModel <- train(y ~ .,
                  data = DATA, 
                  trControl = traincontrol,
                  method = algorithm,
                  metric = "ROC",
                  weights = classWeights,
                  )

, он работает, и ошибки нет. но когда я добавляю аргумент выборки trainControl как

# train parameters
set.seed(12345)
traincontrol <- trainControl(method = "loocv", # resampling method
                             number = 10,
                             index = folds,
                             classProbs = TRUE, 
                             summaryFunction = twoClassSummary,
                             savePredictions = TRUE,
                             sampling = "down"
                             )

fitModel <- train(y ~ .,
                  data = DATA, 
                  trControl = traincontrol,
                  method = algorithm,
                  metric = "ROC",
                  weights = classWeights,
                  )

, я получаю эту ошибку, которая понятна:

Error in model.frame.default(formula = .outcome ~ ., data = list(x = c(-0.0640913631047556,  : 
  variable lengths differ (found for '(weights)')
In addition: There were 11 warnings (use warnings() to see them)
Timing stopped at: 0.112 0.001 0.115

Есть ли способ сделать это в caret? Большое спасибо заранее.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...