R: LIME возвращает ошибку на разных номерах функций, когда это не так - PullRequest
Новая
       26

R: LIME возвращает ошибку на разных номерах функций, когда это не так

0 голосов
/ 10 мая 2018

Я создаю текстовый классификатор твитов Клинтона и Трампа (данные можно найти по Kaggle ).

Я делаю EDA и моделирую, используя пакет quanteda: 

library(dplyr)
library(stringr)
library(quanteda)
library(lime)

#data prep
tweet_csv <- read_csv("tweets.csv")
tweet_data <- tweet_csv %>% 
  select(author = handle,
     text,
     retweet_count,
     favorite_count,
     source_url,
     timestamp = time) %>% 
mutate(date = as_date(str_sub(timestamp, 1, 10)),
     hour = hour(hms(str_sub(timestamp, 12, 19))),
     tweet_num = row_number()) %>% 
select(-timestamp)

# creating corpus and dfm
tweet_corpus <- corpus(tweet_data)

edited_dfm <- dfm(tweet_corpus, remove_url = TRUE, remove_punct = TRUE,     remove = stopwords("english"))

set.seed(32984)
trainIndex <- sample.int(n = nrow(tweet_csv), size =     floor(.8*nrow(tweet_csv)), replace = F)

train_dfm <- edited_dfm[as.vector(trainIndex), ]
train_raw <- tweet_data[as.vector(trainIndex), ]
train_label <- train_raw$author == "realDonaldTrump"

test_dfm <- edited_dfm[-as.vector(trainIndex), ]
test_raw <- tweet_data[-as.vector(trainIndex), ]
test_label <- test_raw$author == "realDonaldTrump"

# making sure train and test sets have the same features
test_dfm <- dfm_select(test_dfm, train_dfm)

# using quanteda's NB model
nb_model <- quanteda::textmodel_nb(train_dfm, train_labels)
nb_preds <- predict(nb_model, test_dfm) 


# defining textmodel_nb as classification model
class(nb_model)

model_type.textmodel_nb_fitted <- function(x, ...) {
  return("classification")
}

# a wrapper-up function for data preprocessing

get_matrix <- function(df){
  corpus <- corpus(df)
  dfm <- dfm(corpus, remove_url = TRUE, remove_punct = TRUE, remove = stopwords("english"))
}

тогда я определяю объяснителя - здесь нет проблем: 

explainer <- lime(train_raw[1:5],
              model = nb_model,
              preprocess = get_matrix)

Но когда я запускаю объяснитель, даже на том же наборе данных, что и в explainer, я получаю ошибку: 

explanation <- lime::explain(train_raw[1:5], 
                              explainer, 
                              n_labels = 1,
                              n_features = 6,
                              cols = 2,
                              verbose = 0)

Ошибка вgnett.textmodel_nb_fitted (x, newdata = newdata, type = type,: набор функций в newdata отличается от этого в обучающем наборе

Это как-то связано с quanteda и dfms? Честно говоря, я не понимаю, почему это должно произойти. Любая помощь будет отличной, спасибо!

1 Ответ

0 голосов
/ 10 мая 2018

Мы можем отследить ошибку до predict_model, что вызывает predict.textmodel_nb_fitted (я использовал только первые 10 строк train_raw для ускорения вычислений): 

traceback()
# 7: stop("feature set in newdata different from that in training set")
# 6: predict.textmodel_nb_fitted(x, newdata = newdata, type = type, 
#        ...)
# 5: predict(x, newdata = newdata, type = type, ...)
# 4: predict_model.default(explainer$model, case_perm, type = o_type)
# 3: predict_model(explainer$model, case_perm, type = o_type)
# 2: explain.data.frame(train_raw[1:10, 1:5], explainer, n_labels = 1, 
#        n_features = 5, cols = 2, verbose = 0)
# 1: lime::explain(train_raw[1:10, 1:5], explainer, n_labels = 1, 
#        n_features = 5, cols = 2, verbose = 0)

Проблема в том, что predict.textmodel_nb_fitted ожидает dfm, а не фрейм данных.Например, predict(nb_model, test_raw[1:5]) выдает ту же ошибку «набор функций в новых данных, отличающийся от ошибки в обучающем наборе».Однако explain принимает в качестве аргумента x фрейм данных.

Решение состоит в том, чтобы написать собственный textmodel_nb_fitted метод для predict_model, который выполняет необходимые преобразования объектов перед вызовом predict.textmodel_nb_fitted: 

predict_model.textmodel_nb_fitted <- function(x, newdata, type, ...) {
  X <- corpus(newdata)
  X <- dfm_select(dfm(X), x$data$x)   
  res <- predict(x, newdata = X, ...)
  switch(
   type,
   raw = data.frame(Response = res$nb.predicted, stringsAsFactors = FALSE),
   prob = as.data.frame(res$posterior.prob, check.names = FALSE)
  )  
}

Это дает нам 

explanation <- lime::explain(train_raw[1:10, 1:5], 
                              explainer,
                              n_labels = 1,
                              n_features = 5,
                              cols = 2,
                              verbose = 0)
explanation[1, 1:5]
#       model_type case label label_prob    model_r2
# 1 classification    1 FALSE  0.9999986 0.001693861
...