R/writeASScenario.R

In aslib: Interface to the Algorithm Selection Benchmark Library

#' @title Writes an algorithm selection scenario to a directory.
#'
#' @description
#' Splits an algorithm selection scenario into description, feature
#' values / runstatus / costs, algorithm performance and cv splits and
#' saves those data sets as single ARFF files in the given directory.
#'
#' @param asscenario [\code{\link{ASScenario}}]\cr
#'   Algorithm selection scenario.
#' @param path [\code{character(1)}]\cr
#'   Path to write scenario to. Default is the name of the scenario.
#' @export
#' @seealso \code{\link{parseASScenario}}
writeASScenario = function(asscenario, path = asscenario$desc$scenario_id) {
  assertClass(asscenario, "ASScenario")

  dir.create(path)
  oldwd = getwd()
  setwd(path)

  desc = lapply(asscenario$desc, yamlify)
  cat(as.yaml(desc), file = "description.txt")

  write.arff(asscenario$feature.values, "feature_values.arff")
	if (!is.null(asscenario$algorithm.feature.values)) {
		write.arff(asscenario$algorithm.feature.values, "algorithm_feature_values.arff")
  }
	write.arff(asscenario$feature.runstatus, "feature_runstatus.arff")
	if (!is.null(asscenario$algorithm.feature.runstatus)) {
  	write.arff(asscenario$algorithm.feature.runstatus, "algorithm_feature_runstatus.arff")
  }
	if (!is.null(asscenario$feature.costs)) {
    write.arff(asscenario$feature.costs, "feature_costs.arff")
  }
  if (!is.null(asscenario$algorithm.feature.costs)) {
    write.arff(asscenario$algorithm.feature.costs, "algorithm_feature_costs.arff")
  }

  if (!is.null(asscenario$algo.runs)) {
    write.arff(asscenario$algo.runs, "algorithm_runs.arff")
  }

  if (!is.null(asscenario$cv.splits)) {
    write.arff(asscenario$cv.splits, "cv.arff")
  }

  setwd(oldwd)
}

yamlify = function(thing) {
  if (length(thing) == 1L && is.na(thing)) {
    "?"
  } else {
    thing
  }
}