giuseppec / giuseppec/iml

Classification tasks are sometimes not recognized

Open
#216 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
R
Stars
503
Forks
87
PR merge metrics
No merged PRs in 30d

Description

Sometimes, `iml:::inferTaskFromPrediction()` function identifies tasks falsely as a `regression` task and then `pred$task = "regression" is falsely set.

Here is an example:
``` r
library(iml)
library(tidymodels)

data(german, package = "rchallenge")
credit = german[, c("duration", "amount", "purpose", "age",
"employment_duration", "housing", "number_credits", "credit_risk")]

# tidymodels
rf = rand_forest(mode = "classification", engine = "randomForest") %>%
fit(credit_risk ~ ., data = credit)

pred = Predictor$new(model = rf, data = credit, y = "credit_risk")
pred$task
#> [1] "unknown"
pred$task = NULL
pred$predict(credit[c(1, 2),])
pred$task
#> [1] "regression"

iml:::inferTaskFromPrediction(prediction = pred$predict(credit[c(1, 2),]))
#> [1] "regression"
```

(This issue was originally raised for the counterfactuals package, see dandls/counterfactuals#29)

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.