[#542] candidate fix for #542

ModelOriented · Jan 15, 2023 · 478a19d · 478a19d
1 parent 7b203c0
commit 478a19d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/NEWS.md b/NEWS.md
@@ -2,6 +2,7 @@ DALEX 2.4.3
 ---------------------------------------------------------------
 * added implementation of aSHAP (aggregated SHAP) and waterfall plot ([#519](https:/ModelOriented/DALEX/issues/519))
 * adding a new system for default color schemes ([#541](https:/ModelOriented/DALEX/issues/541))
+* added `cross_entropy` as model performance measure to multilabel settings [#542](https:/ModelOriented/DALEX/issues/542)
 
 DALEX 2.4.2
 ---------------------------------------------------------------

diff --git a/R/model_performance.R b/R/model_performance.R
@@ -109,7 +109,8 @@ model_performance <- function(explainer, ..., cutoff = 0.5) {
  macro_F1 = model_performance_macro_f1(predicted, observed),
  w_macro_F1 = model_performance_weighted_macro_f1(predicted, observed),
  accuracy = model_performance_accuracy_multi(predicted, observed),
- w_macro_auc = model_performance_weighted_macro_auc(predicted, observed)
+ w_macro_auc = model_performance_weighted_macro_auc(predicted, observed),
+ cross_entropy = loss_cross_entropy(predicted = predicted, observed = observed)
  )
  } else {
  stop("`explainer$model_info$type` should be one of ['regression', 'classification', 'multiclass'] - pass `model_info = list(type = $type$)` to the `explain` function. Submit an issue on https:/ModelOriented/DALEX/issues if you think that this model should be covered by default.")