Update mlClassificationLogisticMultinomial.R

koenderks · koenderks · commit 2986b2ca5b25 · 2024-09-11T00:25:11.000+02:00
diff --git a/R/mlClassificationLogisticMultinomial.R b/R/mlClassificationLogisticMultinomial.R
@@ -128,7 +128,6 @@ mlClassificationLogisticMultinomial <- function(jaspResults, dataset, options, .
     result[["explainer"]] <- DALEX::explain(result[["model"]], type = "classification", data = result[["train"]], y = result[["train"]][, options[["target"]]], predict_function = function(model, data) data.frame(1 - predict(model, newdata = data, type = "response"), predict(model, newdata = data, type = "response")))
     result[["explainer_fi"]] <- DALEX::explain(result[["model"]], type = "classification", data = result[["train"]], y = as.numeric(result[["train"]][, options[["target"]]]) - 1, predict_function = function(model, data) round(predict(model, newdata = data, type = "response"), 0) + 1)
   } else {
-    # TODO
     result[["explainer"]] <- DALEX::explain(result[["model"]][["original"]], type = "multiclass", data = result[["train"]], y = result[["train"]][, options[["target"]]], predict_function = function(model, data) VGAM::predict(model, data, type = "response"))
     result[["explainer_fi"]] <- result[["explainer"]]
   }

Original file line number	Diff line number	Diff line change
`@@ -128,7 +128,6 @@ mlClassificationLogisticMultinomial <- function(jaspResults, dataset, options, .`
`128`	`128`	`result[["explainer"]] <- DALEX::explain(result[["model"]], type = "classification", data = result[["train"]], y = result[["train"]][, options[["target"]]], predict_function = function(model, data) data.frame(1 - predict(model, newdata = data, type = "response"), predict(model, newdata = data, type = "response")))`
`129`	`129`	`result[["explainer_fi"]] <- DALEX::explain(result[["model"]], type = "classification", data = result[["train"]], y = as.numeric(result[["train"]][, options[["target"]]]) - 1, predict_function = function(model, data) round(predict(model, newdata = data, type = "response"), 0) + 1)`
`130`	`130`	`} else {`
`131`		`- # TODO`
`132`	`131`	`result[["explainer"]] <- DALEX::explain(result[["model"]][["original"]], type = "multiclass", data = result[["train"]], y = result[["train"]][, options[["target"]]], predict_function = function(model, data) VGAM::predict(model, data, type = "response"))`
`133`	`132`	`result[["explainer_fi"]] <- result[["explainer"]]`
`134`	`133`	`}`