diff --git a/NEWS.md b/NEWS.md index 466f75964..a9d94ae50 100644 --- 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://github.com/ModelOriented/DALEX/issues/519)) * adding a new system for default color schemes ([#541](https://github.com/ModelOriented/DALEX/issues/541)) +* added `cross_entropy` as model performance measure to multilabel settings [#542](https://github.com/ModelOriented/DALEX/issues/542) DALEX 2.4.2 --------------------------------------------------------------- diff --git a/R/model_performance.R b/R/model_performance.R index 3418c1a0c..b2d62df15 100644 --- 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://github.com/ModelOriented/DALEX/issues if you think that this model should be covered by default.")