In celehs/PETLER: Multi-modal Automated Survival Time Annotation

knitr::opts_chunk$set(echo = TRUE)

library(tidyverse)

Code <- bind_rows(
  Train = read_csv("data_org/TrainCode.csv"), 
  Valid = read_csv("data_org/ValidCode.csv"), .id = "split")

fu_time <- Code %>% select(split, id, fu_time) %>% distinct()
follow_up_train <- fu_time %>% filter(split == "Train") %>% select(-split)
follow_up_valid <- fu_time %>% filter(split == "Valid") %>% select(-split)
write_csv(follow_up_train, "long/follow_up_train.csv")
write_csv(follow_up_valid, "long/follow_up_valid.csv")

for (i in 1:3) {
  pred <- paste0("pred", i)
  path <- paste0("long/time_code", i, ".csv")
  wide <- Code[c(Code[, pred] > 0), c("id", "month", pred)]
  freq <- wide[, pred][[1]]
  long <- tibble(
    id = rep(wide$id, freq),
    month = rep(wide$month, freq))
  write_csv(long, path)  
  print(path)  
}