tests/tests.R

In prescient/opinionated_pipelines: Functions for transaction analytics

# Model tests -------------------------------------------------------------
require(pacman)
p_load(lightgbm,
       vtreat,
       KScorrect,
       plotly,
       yardstick,
       glue,
       parallel,
       doParallel)

#test functions
data(iris)
iris <- iris
target = "Sepal.Length"

mdl_data <- split_data(iris)
mdl_data <- treat_numeric_data(mdl_data   = mdl_data,
                               target_var = "Sepal.Length")
mdl_data <- model_lgb_reg(mdl_data    = mdl_data,
                          target_var  = target,
                          tune_rounds = 50)
plot_hyperparam_tune(mdl_data)
plot_cor_boot(mdl_data)
plot_rmse_boot(mdl_data)
plot_rsq_boot(mdl_data)
plot_mae_boot(mdl_data)
plot_all_metrics(mdl_data)