code/060-modeling/paper/08-train.R

In pat-s/2019-feature-selection: Research Compendium package for the publication "Monitoring forest health using hyperspectral imagery: Does feature selection improve the performance of machine-learning techniques?"

train_plan <- drake_plan(
  train_hr_xgboost = target(
    train(xgboost_borda, task[[1]])
  ),

  iml_new_hr = target(Predictor$new(
    train_hr_xgboost,
    task[[1]]$env$data
  )),

  iml_ale_hr = target({
    cl <- makePSOCKcluster(4)
    registerDoParallel(cl)
    fe <- FeatureEffects$new(iml_new_hr, parallel = TRUE)
    stopCluster(cl)

    return(fe)
  })
)