Skip to content

Commit

Permalink
add convert data to epiweekly to test mode script
Browse files Browse the repository at this point in the history
  • Loading branch information
SamuelBrand1 committed Nov 22, 2024
1 parent 7492852 commit 128ad65
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions pipelines/tests/test_run.sh
Original file line number Diff line number Diff line change
Expand Up @@ -14,22 +14,32 @@ N_AHEAD=$3
# Iterate over each subdirectory in the base directory
echo "TEST-MODE: Running loop over subdirectories in $BASE_DIR"
echo "For $N_SAMPLES samples on 1 chain, and $N_AHEAD forecast points"
echo ""
for SUBDIR in "$BASE_DIR"/*/; do
echo "TEST-MODE: Inference for $SUBDIR"
python fit_model.py "$SUBDIR" --n-chains 1 --n-samples $N_SAMPLES
echo "TEST-MODE: Finished inference"
echo ""
echo "TEST-MODE: Generating posterior predictions for $SUBDIR"
python generate_predictive.py "$SUBDIR" --n-forecast-points $N_AHEAD
echo "TEST-MODE: Finished generating posterior predictions"
echo ""
echo "TEST-MODE: Converting inferencedata to parquet for $SUBDIR"
Rscript convert_inferencedata_to_parquet.R "$SUBDIR"
echo "TEST-MODE: Finished converting inferencedata to parquet"
echo ""
echo "TEST-MODE: Generate epiweekly data for $SUBDIR"
Rscript generate_epiweekly.R "$SUBDIR"
echo "TEST-MODE: Finished generating epiweekly data"
echo ""
echo "TEST-MODE: Forecasting baseline models for $SUBDIR"
Rscript timeseries_forecasts.R "$SUBDIR" --n-forecast-days $N_AHEAD --n-samples $N_SAMPLES
echo "TEST-MODE: Finished forecasting baseline models"
echo ""
echo "TEST-MODE: Postprocessing state forecast for $SUBDIR"
Rscript postprocess_state_forecast.R "$SUBDIR"
echo "TEST-MODE: Finished postprocessing state forecast"
echo ""
echo "TEST-MODE: Scoring forecast for $SUBDIR"
Rscript score_forecast.R "$SUBDIR"
echo "TEST-MODE: Finished scoring forecast"
Expand Down

0 comments on commit 128ad65

Please sign in to comment.