Skip to content

Allow plot_features(cases = ) to accept integer indices even when x has rownames #194

Description

@mattansb
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

Metadata

Metadata

Assignees

No one assigned

    Labels

    featurea feature request or enhancement

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions