/
t_forecast.R
75 lines (55 loc) · 1.94 KB
/
t_forecast.R
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
t_forecast <- list(
pre_forecasting,
state_results,
tar_combine(forecast_starts, state_results[["state_forecast_start_tbl"]],
command = dplyr::bind_rows(!!!.x)),
tar_combine(all_state_trajs, state_results[["state_results_traj"]],
command = dplyr::bind_rows(!!!.x), format = "fst"),
tar_target(
backup_trajs,
{
file_path <- paste0(plot_dir, "/trajectories.fst")
fst::write_fst(all_state_trajs, path = file_path, compress = 100)
if(do_upload_trajectories) {
sync_file <- paste0("/home/forecast/mfluxunimelb/clinical_forecasts/trajectories/trajectories_", forecast_name, ".fst")
file.copy(file_path, sync_file, overwrite = TRUE)
}
return(file_path)
},
format = "file",
deployment = "main"
),
tar_target(
backup_fc_dates,
{
file_path <- paste0(plot_dir, "/forecast_dates.csv")
forecast_dates %>%
write_csv(file_path)
return(file_path)
},
format = "file",
deployment = "main"
),
tar_combine(all_state_ABC_parameters, state_results[["state_ABC_parameters"]]),
tar_combine(all_state_ABC_diagnostics, state_results[["state_ABC_diagnostics"]]),
tar_target(
ABC_diagnostic_plots,
plot_ABC_diagnostics(all_state_ABC_diagnostics, all_state_ABC_parameters, plot_dir),
format = "file",
deployment = "main"
),
tar_combine(all_state_capacity, state_results[["state_capacity_table"]],
command = dplyr::bind_rows(!!!.x)),
tar_target(
state_capacity_report,
{
file_out <- paste0(plot_dir, "/clinical_capacity_", date_forecasting ,".csv")
all_state_capacity %>%
filter(date == forecast_dates$forecast_horizon - ddays(1)) %>%
select(state, group, multiplier, date, prob = y_adj) %>%
write_csv(file_out)
},
cue = tar_cue_skip(is_retro),
deployment = "main"
)
)