For merging lbm-seq results

merge-lbm-seq.R

@@ -0,0 +1,20 @@
+library(stringr)
+
+args <- commandArgs(trailingOnly = TRUE)
+
+path <- args[1]
+mode <- args[2]
+
+base_folder <- str_remove(string = path, pattern = "/tmp[0-9]*$")
+print(base_folder)
+epoch <- str_extract(string = path, pattern = "(?<=tmp)([0-9]*)")
+
+flist <- list.files(path, pattern = paste0(mode, ".Rds"), full.names = TRUE)
+
+stopifnot("No files corresponding to mode found." = length(flist) > 0)
+
+merged_res <- lapply(flist, readRDS)
+
+names(merged_res) <- paste0("Rank", seq_along(flist) + 1)
+
+saveRDS(merged_res, file.path(base_folder, paste0(mode, "_", epoch, ".Rds")))