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
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions