temp-bikes/script-4-UPPMAX/bikes.R

In ooelrich/oscbvar: Stuff for Bayesian vector autoregressions

aaa <- Sys.time()
slurm_id <- as.numeric(Sys.getenv("SLURM_ARRAY_TASK_ID"))
max_id <- as.numeric(Sys.getenv("MAX_ID"))

# Kör "install package first," se obsidian under HPC
#if (!("oscbvar" %in% installed.packages()[, 1])) {
#    install.packages(
#        "/domus/h1/oelrich/rpkt/oscbvar_1.0.tar.gz",
#        repos = NULL,
#	type = "source"
#    )
#}
library(oscbvar)


############################################################
### GENERATE A POOLING FRAME                             ###
############################################################

sotw_all <- cbind(
    subset(bikes_d_log, select = c(t, temp, hum, windspeed))
)

# Matchningsvariabler. Family day är thanks giving, juldag, julafton
matchi <- data.frame(t = 1:730, family_day = 0)
matchi[c(327, 357:358, 691, 723:724), 2] <- 1

#############################################################
### GENERATE BASELINE PREDS AND A COLLECTION OF DIFFERENT ###
### CALIPER AGGREGATIONS                                  ###
#############################################################

# only create baseline values for one of the runs

if(slurm_id == 1) {
    df_agg_base <- gen_agg_preds(
    	atomic_df = bikes_atom,
    	start_agg = 401,
    	sotw = sotw_all,
    	baseline = TRUE,
    	caliper = FALSE,
    	mahala = FALSE,
    	cw = 0.01,
    	mvc = 1
    )

    # add a cw column that is just NA for the baseline
    df_agg_base <- cbind(df_agg_base, calw = NA)
    df_all <- df_agg_base
} else {
   df_all <- data.frame()
}

cw <- 2 * slurm_id / max_id
df_agg <- gen_agg_preds(
    atomic_df = bikes_atom,
    start_agg = 401,
    sotw = sotw_all,
    baseline = FALSE,
    caliper = TRUE,
    mahala = FALSE,
    cw = cw,
    mvc = 1,
    matching_vars = matchi
)
df_agg <- cbind(df_agg, calw = cw)
df_all <- rbind(df_all, df_agg)
colnames(df_all)[6] <- "cw" #  rename the calw-column for consistency

saveRDS(df_all, file = sprintf("/proj/dennis/nobackup/bikes_cw_%s.Rds", slurm_id))
Sys.time() - aaa