WMATA-R/_targets.R at main · ugcodrr/WMATA-R · GitHub

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
library(targets)
library(tarchetypes)

source("functions.R")
source("data_prep.R")
source("feature_engineering.R")
source("model_spec.R")
source("forecasting.R")
source("evaluation.R")
source("model_fit.R")
source("graph_pipeline.R")
source("discovery_layer.R")

tar_option_set(
  packages = project_packages(),
  seed = project_seed(),
  format = "rds"
)

list(
  tar_target(raw_exports, locate_raw_exports()),
  tar_target(training_exclusion_file, here::here("data/training_exclusions.csv"), format = "file"),
  tar_target(training_exclusions, read_training_exclusions(training_exclusion_file)),
  tar_target(bronze_data, build_bronze_data(raw_exports)),
  tar_target(bronze_files, write_bronze_data(bronze_data), format = "file"),
  tar_target(silver_data, build_silver_data(bronze_data)),
  tar_target(silver_files, write_silver_data(silver_data), format = "file"),
  tar_target(station_completeness, build_station_completeness(silver_data$ridership_rail_station_daily)),
  tar_target(gold_data, build_gold_data(silver_data, station_completeness, training_exclusions)),
  tar_target(gold_files, write_gold_data(gold_data), format = "file"),
  tar_target(bus_results, run_bus_pipeline(gold_data$bus_model_frame)),
  tar_target(rail_results, run_rail_pipeline(gold_data, station_completeness, silver_data)),
  tar_target(discovery_results, build_discovery_results(bus_results, rail_results, gold_data, silver_data)),
  tar_target(discovery_table_files, write_discovery_tables(discovery_results), format = "file"),
  tar_target(discovery_figure_files, write_discovery_figures(discovery_results), format = "file"),
  tar_target(discovery_artifact_file, write_discovery_artifact(discovery_results), format = "file"),
  tar_target(table_files, write_pipeline_tables(bus_results, rail_results, silver_data), format = "file"),
  tar_target(
    figure_files,
    write_pipeline_figures(bus_results, rail_results, gold_data$bus_model_frame, silver_data),
    format = "file"
  ),
  tar_target(artifact_file, write_diagnostic_artifacts(bus_results, rail_results), format = "file")
)