inst/do_tune.R

In TatsumotoH/abcihelpr: helper functions for using ABCI (AI Bridging Cloud Infrastructure)

#チューニングのためだけのRスクリプト


library(tidymodels)
library(embed)


#source("RecipeSteps.R")
#source("AdditionalTunes.R")

#tune_wf, tune_folds, param_gridがロードされる
#load("./.param/tune.RData")

#cat(getwd())

args = commandArgs(trailingOnly = T)

f_grid_tune_id = args[1]
grid_tune_start = args[2]
grid_tune_end = args[3]

cat(paste0("f_grid_tune_id=", f_grid_tune_id, "¥n"))
cat(paste0("grid_tune_start=", grid_tune_start, "¥n"))
cat(paste0("grid_tune_end=", grid_tune_end, "¥n"))


param_file = paste0("./params/tune_", f_grid_tune_id, ".Rdata")

#load("tune.Rdata")
#load(param_file)

saved_obj = readRDS(param_file)

grid_tune_id = saved_obj$grid_tune_id
tune_wf = saved_obj$tune_wf
tune_folds = saved_obj$tune_folds
pkg_names = saved_obj$pkg_names
param_grid = saved_obj$param_grid


#load functions from envriontment variable e1
e1 = saved_obj$e1
for(n in ls(e1, all.names = T)){ assign(n, get(n, e1), globalenv())}


#load additional packages
for(pkg_name in pkg_names) {
  if (!require(pkg_name, character.only = TRUE)) {
    install.packages(pkg_name)
    require(pkg_name, character.only = TRUE)
  }
}



# set all the param_grid to target if grid_tune is -1
if( grid_tune_end == -1){
  n_param_grid = dim(param_grid)[1]
  grid_tune_end = n_param_grid
}

cat("set trg_grid_param\n")
cat(paste0("grid_tune_start=", grid_tune_start), "¥n")
cat(paste0("grid_tune_end=", grid_tune_end, "¥n"))

#extract target grid_param
trg_grid_param = param_grid[grid_tune_start:grid_tune_end, ]


#for parallel processing
doParallel::registerDoParallel()


tune_res = tune_grid(
  tune_wf,
  resamples = tune_folds,             #tuneするときに使うデータセット
  grid = trg_grid_param,               #グリッド計画
  control = control_grid(save_workflow = TRUE)
)

timestr = format(Sys.time(), "%H%M%OS")

saveRDS(
  tune_res,
  file = paste0("./output/tune_res_",
                grid_tune_id,
                "_",
                timestr,
                ".obj")
)