Allow `plot_features(cases = )` to accept integer indices even when `x` has rownames
mattansb opened this issue · comments
library(lime)
library(MASS)
iris_test <- iris[1, 1:4]
iris_train <- iris[-1, 1:4]
iris_lab <- iris[[5]][-1]
model <- lda(iris_train, iris_lab)
explanation <- lime(iris_train, model)
explanations <- explain(iris_test, explanation, n_labels = 1, n_features = 2)
plot_features(explanations, cases = 1)
When the data frame X to be explained has rownames, cases
must be a character. It would be great if cases
would also accept integer indices here:
rownames(iris_test) <- "flower"
explanations <- explain(iris_test, explanation, n_labels = 1, n_features = 2)
plot_features(explanations, cases = 1)
#> Error: No explanations to plot
plot_features(explanations, cases = "flower")
Created on 2023-11-12 with reprex v2.0.2