main_R_file.R

In rshudde/airline_GP_prediction: What the package does (short line)

#!/usr/bin/env Rscript
args = commandArgs(trailingOnly=TRUE)
source('R/FUNC_woodchan_samples.R')
source('R/FUNC_paramater_estimates.R')
source('R/DATA_generate_simulation.R')
source('R/FUNC_Gibbs_Sampler.R')
source('R/FUNC_Gibbs_Sampler_r.R')
source('R/PLOTS_Gibbs_sampler.R')
Rcpp::sourceCpp('src/FUNC_paramater_estimates_c.cpp')

# things passed in 
# r <- as.numeric(Sys.getenv("SLURM_ARRAY_TASK_ID")) 
# current_t = ""

# r = as.numeric(args[1])
# current_t = as.numeric(args[2])

##args is now a list of character vectors
## First check to see if arguments are passed.
## Then cycle through each element of the list and evaluate the expressions.
args=(commandArgs(TRUE))
if(length(args)==0){
  print("No arguments supplied.")
  ##supply default values
  stop("NO COMMAND LINE ARGUMENTS PASSED FOR R AND T")
}else{
  for(i in 1:length(args)){
    eval(parse(text=args[[i]]))
  }
}

USE_NNGP = ifelse(USE_NNGP == "TRUE", TRUE, FALSE)
# function to read data
read_data = function(i, t_val){
  X_initial = read.csv(file = paste0("t", t_val, "/X_data_n100_t", t_val, "_rep",i,".csv"), header = T)
  Y = as.matrix(read.csv(file = paste0("t", t_val, "/Y_data_n100_t", t_val, "_rep",i,".csv"), header = T))
  X = list()
  range = nrow(X_initial)/t_val
  k = 1
  for(j in 1:range){
    X[[j]] = as.matrix(X_initial[(k:(j*t_val)), ])
    k = (j*t_val)+1
    
  }
  
  return(list("X" = X, "y" = Y, "t_val" = t_val))
}

data = read_data(r, t)
print(paste("GOT DATA FOR CASE", t, "/", r, "OF DIMENSION", nrow(data$X[[1]]), ",", ncol(data$X[[1]])))

# run the simluations
results = gibbs_sampler_r(data_gibbs = data, 
                          B = B_VAL,
                          xi_initial = runif(data$t_val, -1, 1),
                          burn_in = 0.5,
                          NNGP = USE_NNGP,
                          n_to_store = STORE_VAL)


if (USE_NNGP)
{
  filename = paste("RESULTSNNGP/results_n100_t", t, "_rep", r, "_NNGP.rda", sep = "")
} else {
  filename = paste("RESULTS/results_n100_t", t, "_rep", r, ".rda", sep = "")
}
save(results, file = filename)