Imports

knitr::opts_chunk$set(echo = FALSE)
h2o::h2o.no_progress()
# Script parameters
devtools::load_all()
database <- "test_signauxfaibles"
collection <- "Features"
actual_period <- as.Date("2019-03-01")
last_batch <- "1905_repeatable"
min_effectif <- 10
retrain_model <- TRUE
export_type <- c("csv", "mongodb")
sample_size <- 800000
connect_to_h2o()
task <- sf_task(
  database,
  collection,
  id = "Reference evaluation canvas"
)

task <- load_hist_data(
  task = task,
  batch = last_batch, subsample = sample_size)

task <- split_n_folds(task, 4, 0.2)
task <- prepare(task, data_names = c("train_data", "validation_data"))
# task <- optimize_hyperparameters(task, n_iter = 10)
 task[["model_parameters"]] <- list(
   learn_rate = 0.044612,
   max_depth = 7,
   ntrees = 237,
   min_child_weight = 3
   )
task <- train(task)
task <- predict(task, data_names = "validation_data")
task <- evaluate(task, data_name = "validation_data")
task <- log(task)

OU

task <- evaluate(task, data_name = "validation_data", eval_function = custom_eval_urssaf())


signaux-faibles/rsignauxfaibles documentation built on Dec. 2, 2020, 3:24 a.m.