Visualization functions and Shiny app

DESCRIPTION

@@ -4,7 +4,8 @@ Version: 0.0.1
 Authors@R: c(
     person("Damir", "Pulatov", , "[email protected]", role = c("cre", "aut")),
     person("Marc", "Becker", , "[email protected]", role = "aut",
-           comment = c(ORCID = "0000-0002-8115-0400"))
+           comment = c(ORCID = "0000-0002-8115-0400")),
+    person("Baisu", "Zhou", , "[email protected]", role = "aut")
   )
 Description: Flexible AutoML system for the 'mlr3' ecosystem.
 License: LGPL-3
@@ -18,15 +19,21 @@ Depends:
     R (>= 3.1.0),
     rush
 Imports:
+    bbotk,
+    bslib,
     checkmate,
     data.table,
+    ggplot2,
+    iml,
     lhs,
     mlr3mbo,
     mlr3misc (>= 0.15.1),
     mlr3pipelines,
     mlr3tuningspaces,
     paradox (>= 1.0.1),
     R6,
+    shiny,
+    stats,
     utils
 Suggests:
     catboost,
@@ -41,7 +48,8 @@ Suggests:
     ranger,
     rpart,
     testthat (>= 3.0.0),
-    xgboost
+    xgboost,
+    vdiffr
 Remotes:
     catboost/catboost/catboost/R-package,
     mlr-org/mlr3,
@@ -76,6 +84,9 @@ Collate:
     'build_graph.R'
     'estimate_memory.R'
     'helper.R'
+    'helpers_app.R'
     'internal_measure.R'
     'train_auto.R'
+    'visualization.R'
+    'visualization_app.R'
     'zzz.R'

NAMESPACE

@@ -27,10 +27,16 @@ export(LearnerRegrAutoNnet)
 export(LearnerRegrAutoRanger)
 export(LearnerRegrAutoSVM)
 export(LearnerRegrAutoXgboost)
+export(cost_over_time)
 export(estimate_memory)
 export(internal_measure_catboost)
 export(internal_measure_lightgbm)
 export(internal_measure_xgboost)
+export(marginal_plot)
+export(parallel_coordinates)
+export(pareto_front)
+export(partial_dependence_plot)
+export(visualize)
 import(R6)
 import(checkmate)
 import(data.table)

R/helpers_app.R

@@ -0,0 +1,42 @@
+#' @title Custom conditionalPanel for hyperparameter selection
+#' 
+#' @description
+#' Used for Marginal Plots and Partial Dependence Plots.
+#' 
+#' @param condition (`character(1)`)\cr
+#'   Passed to the `condition` argument of `[shiny::conditionalPanel]`.
+#' @param prefix (`character(1)`)\cr
+#'   Prefix of input slot names.
+#' @param learner_ids (`character()`)\cr
+#'   Vector of all possible learner/branch IDs.
+#' @param param_ids (`character()`)\cr
+#'   Vector of all possible param IDs.
+#' @param ... (anything)
+#'   Additional arguments passed to `[shiny::conditionalPanel]`.
+#' 
+param_panel = function(condition, prefix, learner_ids, param_ids, ...) {
+  assert_string(condition)
+  assert_string(prefix)
+  assert_character(learner_ids)
+  assert_character(param_ids)
+
+  shiny::conditionalPanel(
+    condition,
+    shiny::selectInput(paste0(prefix, "_branch"),
+      label = "Select branch:",
+      choices = learner_ids
+    ),
+    # choices and selected are just placeholders for initialization
+    shiny::selectInput(paste0(prefix, "_x"),
+      label = "Select x-axis:",
+      choices = param_ids,
+      selected = param_ids[[1]]
+    ),
+    shiny::selectInput(paste0(prefix, "_y"),
+      label = "Select y-axis:",
+      choices = param_ids,
+      selected = param_ids[[2]]
+    ),
+    ...
+  )
+}