#' Expand Fairness Object
#'
#' Unfold fairness object to 3 columns (metrics, label, score) to construct better base for visualization.
#'
#' @param x object of class \code{fairness_object}
#' @param drop_metrics_with_na logical, if \code{TRUE} metrics with NA will be omitted
#' @param scale logical, if \code{TRUE} standardized.
#' @param fairness_metrics character, vector of fairness metrics names indicating from which expand.
#'
#' @export
#' @rdname expand_fairness_object
#' @return object of class \code{expand_fairness_object}. It is a \code{data.frame} with scores for each metric and model.
#'
#'
#' @examples
#'
#' data("german")
#'
#' y_numeric <- as.numeric(german$Risk) - 1
#'
#' lm_model <- glm(Risk ~ .,
#' data = german,
#' family = binomial(link = "logit")
#' )
#'
#' explainer_lm <- DALEX::explain(lm_model, data = german[, -1], y = y_numeric)
#'
#' fobject <- fairness_check(explainer_lm,
#' protected = german$Sex,
#' privileged = "male"
#' )
#' expand_fairness_object(fobject, drop_metrics_with_na = TRUE)
#' \donttest{
#' rf_model <- ranger::ranger(Risk ~ .,
#' data = german,
#' probability = TRUE,
#' num.trees = 200
#' )
#'
#'
#' explainer_rf <- DALEX::explain(rf_model, data = german[, -1], y = y_numeric)
#'
#' fobject <- fairness_check(explainer_rf, fobject)
#'
#' expand_fairness_object(fobject, drop_metrics_with_na = TRUE)
#' }
#'
expand_fairness_object <- function(x, scale = FALSE, drop_metrics_with_na = FALSE, fairness_metrics = NULL) {
stopifnot(is.logical(scale))
stopifnot(is.logical(drop_metrics_with_na))
stopifnot(class(x) == "fairness_object")
n_exp <- length(x$explainers)
parity_loss_metric_data <- x$parity_loss_metric_data
labels <- x$label
if (!is.null(fairness_metrics)) {
parity_loss_metric_data <- parity_loss_metric_data[fairness_metrics]
}
if (drop_metrics_with_na) {
parity_loss_metric_data <- drop_metrics_with_na(parity_loss_metric_data)
}
if (scale) parity_loss_metric_data <- as.data.frame(scale(as.matrix(parity_loss_metric_data)))
# rows = metrics * explainers
expanded_data <- data.frame()
column_names <- colnames(parity_loss_metric_data)
n_metrics <- ncol(parity_loss_metric_data)
for (i in seq_len(n_metrics)) {
to_add <- data.frame(
metric = rep(column_names[i], n_exp),
model = labels,
score = parity_loss_metric_data[, i]
)
expanded_data <- rbind(expanded_data, to_add)
}
expanded_data$metric <- as.factor(expanded_data$metric)
expanded_data$model <- as.factor(expanded_data$model)
expanded_data$score <- as.numeric(expanded_data$score)
rownames(expanded_data) <- NULL
return(expanded_data)
}
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.